Variational Inference Python Packages

pyro-ppl

Deep universal probabilistic programming with Python and PyTorch

pymc

Bayesian Modeling and Probabilistic Programming in Python

pymc3

Bayesian Modeling and Probabilistic Programming in Python

gpflow

Gaussian processes in TensorFlow

normflows

PyTorch implementation of normalizing flow models

probreg

Python package for point cloud registration using probabilistic model (Coherent Point Drift, GMMReg, SVR, GMMTree, FilterReg, Bayesian CPD)

scdef

Deep exponential families for single-cell data.

linfa-vi

A Python library for variational inference with normalizing flow and annealing

torchebm

Components and algorithms for energy-based models

viprs

Variational Inference of Polygenic Risk Scores

pymc-nightly

Probabilistic Programming in Python: Bayesian Modeling and Probabilistic Machine Learning with Aesara

tuotuo

TuoTuo is a Topic Modeling library for Researchers and Engineers

avici

Amortized Inference for Causal Structure Learning

vireosnp

Demultiplexing pooled scRNA-seq data with or without genotype reference

torchtree

Phylogenetic inference with pytorch

move-dl

Multi-omics variational autoencoder

bayestorch

Lightweight Bayesian deep learning library for fast prototyping based on PyTorch

torchbayesian

PyTorch-native Bayesian neural networks for uncertainty quantification in deep learning

mxfusion

Modular Probabilistic Programming on MXNet

dibs-lib

DiBS: Differentiable Bayesian Structure Learning

vistan

Library to run VI algorithms on Stan models.

ivon-opt

IVON optimizer for neural networks based on variational learning.

viabel

Efficient, lightweight variational inference and approximation bounds

mixmil

Code for the paper: Mixed Models with Multiple Instance Learning