Normalizing Flows PyTorch ML Frameworks

PyTorch implementations of normalizing flows and flow-matching algorithms for density estimation and generative modeling. Does NOT include general generative models (diffusion, VAEs, GANs), reinforcement learning with flows, or physics simulations.

There are 30 normalizing flows pytorch frameworks tracked. 1 score above 70 (verified tier). The highest-rated is probabilists/zuko at 75/100 with 446 stars and 26,000 monthly downloads.

Get all 30 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=normalizing-flows-pytorch&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	probabilists/zuko Normalizing flows in PyTorch	75	Verified	446	Python
2	lucidrains/rectified-flow-pytorch Implementation of rectified flow and some of its followup research /...	69	Established	426	Python
3	davidnabergoj/torchflows Modern normalizing flows in Python. Simple to use and easily extensible.	51	Established	12	Python
4	keishihara/flow-matching Flow Matching implemented in PyTorch	50	Established	99	Python
5	LukasRinder/normalizing-flows Implementation of normalizing flows in TensorFlow 2 including a small tutorial.	49	Emerging	148	Jupyter Notebook
6	janosh/awesome-normalizing-flows Awesome resources on normalizing flows.	47	Emerging	1,611	Python
7	Coldog2333/pytoflow The py version of toflow → https://github.com/anchen1011/toflow	45	Emerging	111	MATLAB
8	ikostrikov/pytorch-flows PyTorch implementations of algorithms for density estimation	45	Emerging	586	Python
9	abdulfatir/normalizing-flows Understanding normalizing flows	43	Emerging	132	Jupyter Notebook
10	kamenbliznashki/normalizing_flows Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF,...	40	Emerging	637	Python
11	scikit-mobility/DeepGravity a PyTorch implementation of the paper "Deep Gravity: enhancing mobility...	39	Emerging	115	Jupyter Notebook
12	senya-ashukha/real-nvp-pytorch Real NVP PyTorch a Minimal Working Example \| Normalizing Flow	38	Emerging	142	Jupyter Notebook
13	TrentBrick/PyTorchDiscreteFlows Discrete Normalizing Flows implemented in PyTorch	32	Emerging	115	Jupyter Notebook
14	stevenygd/NFGP Pytorch implementation of NeurIPS 2021 paper: Geometry Processing with Neural Fields.	32	Emerging	204	Jupyter Notebook
15	Efficient-Scalable-Machine-Learning/event-ssm Official implementation of our paper "Scalable Event-by-event Processing of...	31	Emerging	23	Python
16	fangzr/SSM-PC The code repository for our paper "Shared Spatial Memory Through Predictive Coding".	31	Emerging	12	Python
17	buettnerlab/convexgating Optimal gating strategies for flow and mass cytometry.	30	Emerging	16	Python
18	mathisgerdes/continuous-flow-lft Continuous normalizing flow for lattice quantum field theory	29	Experimental	13	Jupyter Notebook
19	DanielEftekhari/normality-normalization Code accompanying the ICML 2025 paper "On the Importance of Gaussianizing...	29	Experimental	6	Python
20	UditBhaskar19/GRAPH_NEURAL_NETWORK_FOR_RADAR_PERCEPTION Multi-task learning using message passing graph neural network for radar...	28	Experimental	8	Jupyter Notebook
21	d-tiapkin/gflownet-rl Repository for "Generative Flow Networks as Entropy-Regularized RL"...	26	Experimental	40	Python
22	minaskar/sinflow Sliced Iterative Normalizing Flow with Minimal Dependencies	23	Experimental	11	Python
23	Candibulldog/MeanFlow-Edge2Image Unofficial PyTorch implementation of "Mean Flows for One-step Generative...	22	Experimental	3	Python
24	volagold/fce-2d Flow Contrastive Estimation (FCE) PyTorch Implementation on 2D data	21	Experimental	11	Python
25	tgritsaev/gflownet-tlm The source code for the paper "Optimizing Backward Policies in GFlowNets via...	20	Experimental	27	Python
26	yixuan/temperflow Efficient Multimodal Sampling via Tempered Distribution Flow	18	Experimental	11	Python
27	honglinwen/Conditional-normalizing-flow-for-wind-power-forecasting Code for paper "Continuous and Distribution-free Probabilistic Wind Power...	18	Experimental	22	Jupyter Notebook
28	EricZhang1412/Spatial-temporal-ERF Official repo for NeurIPS 2025 poster: Unveiling the Spatial-temporal...	16	Experimental	12	Python
29	Boketto-rgb/min-pi-flow 🔍 Implement flow matching for efficient few-step generation with this...	14	Experimental	—	Python
30	aliutkus/groupmap GroupMap: beyond mean and variance matching for deep learning	14	Experimental	11	Python

Comparisons in this category

zuko and normalizing_flows (75 vs 40) pytorch-flows and PyTorchDiscreteFlows (45 vs 32) torchflows and normalizing_flows (51 vs 40) torchflows and pytorch-flows (51 vs 45) zuko and normalizing-flows (75 vs 49) zuko and normalizing-flows (75 vs 43) zuko and PyTorchDiscreteFlows (75 vs 32) rectified-flow-pytorch and flow-matching (69 vs 50) zuko and torchflows (75 vs 51) torchflows and normalizing-flows (51 vs 49)