karpathy / pytorch-normalizing-flows

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

845 stars 98 forks source link

readme

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?