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

Readme.md

pytorch-normalizing-flows

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

Normalizing Flow fitting a 2D dataset

todos

  • 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.