/pytorch-normalizing-flows-AF-only

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

Primary LanguageJupyter Notebook

pytorch-normalizing-flows

This is a simplified PyTorch code of MAF and IAF from Karpathy.