jax-flows

github.comtool

Normalizing flows in JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   NuX github.com

   tool category #jax link ↗

   Normalizing flows with JAX.

   
2. ▲
   flowjax github.com

   tool category #jax link ↗

   Distributions and normalizing flows built as equinox modules.

3. ▲
   Normalizing Flows in 100 Lines of JAX by Eric Jang blog.evjang.com

   site category #jax link ↗

   Concise implementation of RealNVP.

   
4. ▲
   StateFlow github.com

   tool category #clojure link ↗
   
5. ▲
   jax-raft github.com

   tool category #jax link ↗

   JAX/Flax port of the RAFT optical flow estimator.

   
6. ▲
   TF2JAX github.com

   tool category #jax link ↗

   Convert functions/graphs to JAX functions.

   

JAX › Libraries > New Libraries: “Normalizing flows in JAX.”