FOX (FlOws in JAX) WIP

Normalizing flow implementations based on flax.

Tests

Run tests with:

 python -m pytest tests

Current examples in the examples directory:

Name		Name	Last commit message	Last commit date
Latest commit History 22 Commits
examples		examples
src/fox		src/fox
tests		tests
.gitignore		.gitignore
README.md		README.md
poetry.lock		poetry.lock
pyproject.toml		pyproject.toml
setup.py		setup.py