https://github.com/ChrisWaites/jax-flows
Python259
11 months ago
Normalizing Flows in JAX 🌊
MIT License