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

viprs

Variational Inference of Polygenic Risk Scores

ptgp

GPs in PyTensor

sushie

Software to perform multi-ancestry SNP fine-mapping on molecular data

perturbvi

Perturbvi is a scalable approach to infer regulatory modules in large-scale single-cell pertubation screen data.

pyaif-toolkit

PyAIF (pyaif-toolkit): active inference in Python with categorical and continuous observations, shallow inference, and deep temporal inference.

linfa-vi

A Python library for variational inference with normalizing flow and annealing

scdef

Deep exponential families for single-cell data.

vireosnp

Demultiplexing pooled scRNA-seq data with or without genotype reference

mxfusion

Modular Probabilistic Programming on MXNet

avici

Amortized Inference for Causal Structure Learning, NeurIPS 2022

pymc-nightly

Bayesian Modeling and Probabilistic Programming in Python

tuotuo

TuoTuo is a Topic Modeling library for Researchers and Engineers

move-dl

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

dibs-lib

DiBS: Differentiable Bayesian Structure Learning, NeurIPS 2021

ivon-opt

IVON optimizer for neural networks based on variational learning.

viabel

Efficient, lightweight variational inference and approximation bounds

torchtree

A probabilistic framework in PyTorch for phylogenetic models

vistan

A simple library to run variational inference on Stan models.