PyPI page
Home page
Author:
None
Summary:
Bijections & normalizing flows with JAX/NNX
Latest version:
1.2.1
Required dependencies:
chex
|
diffrax
|
einops
|
flax
|
jax
|
jax-autovmap
|
jaxlib
|
numpy
|
optax
Optional dependencies:
black
|
flowjax
|
furo
|
hypothesis
|
ipykernel
|
ipywidgets
|
isort
|
matplotlib
|
mypy
|
myst-nb
|
nb-clean
|
numpy
|
numpydoc
|
pre-commit
|
pytest
|
pytest-cov
|
pytest-xdist
|
ruff
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-copybutton
|
sphinx-math-dollar
Downloads last day:
16
Downloads last week:
41
Downloads last month:
83