Normalizing-flow enhanced sampling package for probabilistic inference in Jax

Normalizing Flows in JAX 🌊

Denoising Diffusion Implicit Models

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Generative Flow Networks - GFlowNet

This is a Pytorch implementation of [normalizing flows on tori and spheres, ICML 2020]

JAX exponential map normalising flows on sphere

Riemannian Convex Potential Maps

Normalizing flows in PyTorch

Code for Neural Spline Flows paper