Normalizing Flows in 100 Lines of JAX by Eric Jang

blog.evjang.comsite

Concise implementation of RealNVP.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   NuX github.com

   tool category #jax link ↗

   Normalizing flows with JAX.

   
2. ▲
   jax-flows github.com

   tool category #jax link ↗

   Normalizing flows in JAX.

   
3. ▲
   flowjax github.com

   tool category #jax link ↗

   Distributions and normalizing flows built as equinox modules.

4. ▲
   Plugging Into JAX by Nick Doiron medium.com

   site category #jax link ↗

   Compares Flax, Haiku, and Objax on the Kaggle flower classification challenge.

5. ▲
   Achieving 4000x Speedups with PureJaxRL chrislu.page

   site category #jax link ↗

   A blog post on how JAX can massively speedup RL training through vectorisation.

6. ▲
   Flax NNX github.com

   site category #jax link ↗

   An evolution on Flax by the same team

JAX › Tutorials and Blog Posts: “Concise implementation of RealNVP.”