Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 593 Bytes

Readme.md

File metadata and controls

13 lines (10 loc) · 593 Bytes

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?