Bayesian Inference Python Packages

pytensor

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

pymc

Bayesian Modeling and Probabilistic Programming in Python

numpyro

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

pyro-ppl

Deep universal probabilistic programming with Python and PyTorch

pymc3

Bayesian Modeling and Probabilistic Programming in Python

blackjax

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

dynesty

Dynamic Nested Sampling package for computing Bayesian posteriors and evidences

bambi

BAyesian Model-Building Interface (Bambi) in Python.

antecedent

Antecedent is a typed causal-analysis compiler whose output is a qualified scientific claim. It keeps identification, structural uncertainty, empirical support, statistical uncertainty, assumptions, implementation evidence and provenance separate and machine-checkable all the way through execution and serialization.

cpnest

Parallel nested sampling

snowline

Fit and compare complex models quickly. Laplace Approximation, Variational Bayes, Importance Sampling.

autofit

PyAutoFit: Classy Probabilistic Programming

bilby

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

autolens

PyAutoLens: Open-Source Strong Gravitational Lensing

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.

bnlearn

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

cobaya

Code for Bayesian Analysis

anesthetic

Nested Sampling post-processing and plotting

nessai

nessai: Nested Sampling with Artificial Intelligence

scientific-computing-system-2-0

Scientific computing platform on NumPy/SciPy/pandas/matplotlib: linear algebra, stats, optimization, signals, ML, information theory, chaos, Bayes, metaheuristics, geometry and RL.

uqa

Embeddable Rust query engine unifying relational, full-text, vector, graph, and probabilistic retrieval under one typed algebra.

gpjax

Gaussian processes in JAX and Equinox.

pyhgf

PyHGF: A neural network library for predictive coding

nestle

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