GitHub repository: ChrisWaites/jax-flows
Primary language: Python
Description: Normalizing Flows in JAX 🌊