karpathy/pytorch-normalizing-flows Normalizing flows in PyTorch. Current intended use is education not production.


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

TODO make this into a proper repo, for now just see the notebooks.


  • TODO: make this a proper repository
  • TODO: make work on GPU
  • TODO: 2D -> ND: get (flat) MNIST to work, reproduce numbers
  • TODO: ND -> images (multi-scale architectures, Glow nets, etc)
  • 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?