Skip to content

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

Notifications You must be signed in to change notification settings

karpathy/pytorch-normalizing-flows

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published