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

scdef

Deep exponential families for single-cell data.

linfa-vi

A Python library for variational inference with normalizing flow and annealing

probreg

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

viprs

Variational Inference of Polygenic Risk Scores (VIPRS)

pymc-nightly

Bayesian Modeling and Probabilistic Programming in Python

torchebm

Components and algorithms for energy-based models

move-dl

MOVE (Multi-Omics Variational autoEncoder) for integrating multi-omics data and identifying cross modal associations

vireosnp

vireoSNP - donor deconvolution for multiplexed scRNA-seq data

avici

Amortized Inference for Causal Structure Learning, NeurIPS 2022

tuotuo

LDA & Neura based topic modelling library

torchtree

A probabilistic framework in PyTorch for phylogenetic models

vistan

Library to run VI algorithms on Stan models.

dibs-lib

DiBS: Differentiable Bayesian Structure Learning, NeurIPS 2021

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

ivon-opt

An optimizer for neural networks based on variational learning

viabel

Efficient, lightweight variational inference and approximation bounds

mixmil

Attention-based Multi-instance Mixed Models