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

4 Commits
 
 
 
 
 
 

Repository files navigation

pytorch-normalizing-flows

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

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

todos

  • 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?

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published