Normalizing Flows Python Packages

zuko

Normalizing flows in PyTorch

98K 466 40

nflows

Normalizing flows in PyTorch

76K 1K 138

glasflow

Normalising flows implemented using nflows

15K 27 4

nessai

nessai: Nested Sampling with Artificial Intelligence

14K 50 4

avitai-artifex

Generative modeling for JAX/Flax NNX: VAEs, GANs, DDPM/DDIM/score/DiT/latent diffusion, normalizing flows, energy-based, autoregressive and geometric (SE(3), protein, point cloud, mesh) models; image, text, audio, molecular, protein, tabular and time-series modalities; a shared Trainer with callbacks and extensions; evaluation through calibrax.

10K 1 0

lightgbmlss

An extension of LightGBM to probabilistic modelling

5K 397 38

pyknos

Conditional density estimation with neural networks

4K 35 6

flowjax

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

4K 244 25

xgboostlss

An extension of XGBoost to probabilistic modelling

4K 730 86

umnn

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

2K 129 21

raylab

Reinforcement learning algorithms in RLlib

2K 58 10

pocomc

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

2K 127 12

pythae

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

1K 2K 177

linfa-vi

A Python library for variational inference with normalizing flow and annealing

1K 15 2

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.

826 47 1

neuronstar

NeuronStar: open-source code and reproduction package for "Fast Bayesian Updating of the Neutron-Star Equation of State with Neural Posterior and Evidence Estimation"

773 2 0

lampe

Likelihood-free AMortized Posterior Estimation with PyTorch

709 137 12