Bayesian Inference Python Packages

pyro-ppl

Deep universal probabilistic programming with Python and PyTorch

1.2M 9K 1K

pymc

Bayesian Modeling and Probabilistic Programming in Python

1.1M 10K 2K

pytensor

PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.

1M 609 182

numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

577K 3K 283

pymc3

Bayesian Modeling and Probabilistic Programming in Python

332K 10K 2K

blackjax

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

135K 1K 139

autofit

PyAutoFit: Classy Probabilistic Programming

93K 64 12

dynesty

Dynamic Nested Sampling package for computing Bayesian posteriors and evidences

91K 407 87

bambi

BAyesian Model-Building Interface (Bambi) in Python.

49K 1K 143

bilby

A unified framework for stochastic sampling packages and gravitational-wave inference in Python.

28K 117 130

anesthetic

Nested Sampling post-processing and plotting

27K 71 19

autolens

PyAutoLens: Open-Source Strong Gravitational Lensing

24K 182 36

bnlearn

Python package for Causal Discovery by learning the graphical structure of Bayesian networks. Structure Learning, Parameter Learning, Inferences, Sampling methods.

21K 625 58

nestle

Pure Python, MIT-licensed implementation of nested sampling algorithms for evaluating Bayesian evidence.

18K 76 27

cpnest

Parallel nested sampling

17K 31 25

pyhgf

PyHGF: A neural network library for predictive coding

14K 137 30

zeus-mcmc

⚡️ zeus: Lightning Fast MCMC ⚡️

14K 244 34

sbi

sbi is a Python package for simulation-based inference, designed to meet the needs of both researchers and practitioners. Whether you need fine-grained control or an easy-to-use interface, sbi has you covered.

14K 823 243