Search Packages

nflows

Normalizing flows in PyTorch

zuko

Normalizing flows in PyTorch

glasflow

Normalising flows implemented using nflows

nessai

nessai: Nested Sampling with Artificial Intelligence

flowmc

Normalizing-flow enhanced sampling package for probabilistic inference in JAX

flowjax

Easy to use distributions, bijections and normalizing flows in JAX.

pyknos

Conditional density estimation with neural networks

pythae

Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)

raylab

Reinforcement learning algorithms in RLlib

umnn

Implementation of Unconstrained Monotonic Neural Network and the related experiments. These architectures are particularly useful for modelling monotonic transformations in normalizing flows.

scyan

Biology-driven deep generative model for cell-type annotation in cytometry. Scyan is an interpretable model that also corrects batch-effect and can be used for debarcoding or population discovery.

xgboostlss

An extension of XGBoost to probabilistic modelling

lightgbmlss

An extension of LightGBM to probabilistic modelling

linfa-vi

A Python library for variational inference with normalizing flow and annealing

pocomc

pocoMC: A Python implementation of Preconditioned Monte Carlo for accelerated Bayesian Computation

surjectors

Surjection layers for density estimation with normalizing flows

lampe

Likelihood-free AMortized Posterior Estimation with PyTorch

avitai-artifex

A research-focused modular generative modeling library built on JAX/Flax NNX

sbiax

Fast, lightweight and parallelised simulation-based inference in JAX.

vistan

Library to run VI algorithms on Stan models.

sinflow

Sliced Iterative Normalizing Flow

deeprob-kit

A Python Library for Deep Probabilistic Modeling

probaforms

Conditional normalizing flows (NFs), conditional GANs, and conditional variational autoencoders (CVAEs) with sklearn-like interface

causalflows

CausalFlows: A library for Causal Normalizing Flows in Pytorch