Bayesian Inference Python Packages

pymc

Bayesian Modeling and Probabilistic Programming in Python

3.8M 10K 2K

pytensor

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

1.8M 625 194

pyro-ppl

Deep universal probabilistic programming with Python and PyTorch

1M 9K 1K

numpyro

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

715K 3K 289

pymc3

Bayesian Modeling and Probabilistic Programming in Python

342K 10K 2K

blackjax

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

173K 1K 142

dynesty

Dynamic Nested Sampling package for computing Bayesian posteriors and evidences

100K 414 88

bambi

BAyesian Model-Building Interface (Bambi) in Python.

56K 1K 145

bilby

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

28K 121 137

bnlearn

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

25K 631 59

autofit

PyAutoFit: Classy Probabilistic Programming

24K 65 12

nestle

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

20K 76 27

cpnest

Parallel nested sampling

19K 31 25

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.

19K 843 243

autolens

PyAutoLens: Open-Source Strong Gravitational Lensing

17K 184 36