FOX (FlOws in JAX) WIP Normalizing flow implementations based on flax. Tests Run tests with: python -m pytest tests Examples Current examples in the examples directory: RealNVP