karpathy / pytorch-normalizing-flows

Normalizing flows in PyTorch. Current intended use is education not production.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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?

About

Normalizing flows in PyTorch. Current intended use is education not production.


Languages

Language:Jupyter Notebook 96.7%Language:Python 3.3%