Name		Name	Last commit message	Last commit date
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

Repository files navigation

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

About

Normalizing Flows in JAX

jax machine-learning normalizing-flows python

1 star

1 watching

Report repository

Releases

Packages

Contributors

Languages