Skip to content
Normalizing flows in PyTorch. Current intended use is education not production.
Jupyter Notebook Python
Branch: master
Clone or download
Type Name Latest commit message Commit time
Failed to load latest commit information.
nflib fix bug where net_class was not correctly used and propagated Dec 12, 2019 small updates to Readme Dec 12, 2019
nflib1.ipynb include also and play with neural splines too Dec 12, 2019


Implementions of normalizing flows (NICE, RealNVP, MAF, IAF, Neural Splines Flows, etc) in PyTorch.

Normalizing Flow fitting a 2D dataset


  • TODO: make work on GPU
  • TODO: 2D -> ND: get (flat) using MNIST
  • TODO: ND -> images (multi-scale architectures, Glow nets, etc) on MNIST/CIFAR/ImageNet
  • TODO: more stable residual-like IAF-style updates (tried but didn't work too well)
  • TODO: parallel wavenet
  • TODO: radial/planar 2D flows from Rezende Mohamed 2015?
You can’t perform that action at this time.