Variational Inference Python Packages

pymc

Bayesian Modeling and Probabilistic Programming in Python

pyro-ppl

Deep universal probabilistic programming with Python and PyTorch

pymc3

Bayesian Modeling and Probabilistic Programming in Python

gpflow

Gaussian processes in TensorFlow

probreg

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

normflows

PyTorch implementation of normalizing flow models

ptgp

GPs in PyTensor

scdef

Deep exponential families for single-cell data.

torchebm

🍓 Build and train energy-based and diffusion models in PyTorch ⚡.

linfa-vi

A Python library for inference with normalizing flow and annealing

viprs

Variational Inference of Polygenic Risk Scores

torchtree

A probabilistic framework in PyTorch for phylogenetic models

avici

Amortized Inference for Causal Structure Learning, NeurIPS 2022

pymc-nightly

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

bayestorch

Lightweight Bayesian deep learning library for fast prototyping based on PyTorch

move-dl

Multi-omics variational autoencoder

dibs-lib

DiBS: Differentiable Bayesian Structure Learning

topomics

Topic modeling for multiomics data

vireosnp

vireoSNP - donor deconvolution for multiplexed scRNA-seq data

torchbayesian

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

tuotuo

TuoTuo is a Topic Modeling library for Researchers and Engineers

ivon-opt

An optimizer for neural networks based on variational learning

mixmil

Code for the paper: Mixed Models with Multiple Instance Learning

viabel

Efficient, lightweight variational inference and approximation bounds