Kelvinson / nf-jax

Normalizing Flows in Jax

nf-jax

Normalizing Flows Tutorial in Jax for ICML INNF workshop. Implementation of Real-NVP in just 75 lines of Python!

Slides here: https://docs.google.com/presentation/d/1wHJz9Awhlp-PWLZGWJKzF66gzvqdSrhknb-iLFJ1Owo/edit?usp=sharing