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.

588K 3K 283

pymc3

Bayesian Modeling and Probabilistic Programming in Python

329K 10K 2K

blackjax

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

136K 1K 139

dynesty

Dynamic Nested Sampling package for computing Bayesian posteriors and evidences

93K 407 87

autofit

PyAutoFit: Classy Probabilistic Programming

92K 64 12

bambi

BAyesian Model-Building Interface (Bambi) in Python.

50K 1K 143

bilby

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

29K 117 130

anesthetic

Nested Sampling post-processing and plotting

27K 71 19

autolens

PyAutoLens: Open-Source Strong Gravitational Lensing

22K 182 36

bnlearn

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

22K 625 58

nestle

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

17K 76 27

cpnest

Parallel nested sampling

16K 31 25

pyhgf

PyHGF: A neural network library for predictive coding

14K 137 30

gpjax

Gaussian processes in JAX and Equinox.

14K 618 73

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