chex>=0.1.89
diffrax>=0.6.2
einops>=0.8.0
flax>=0.12.0
jax>=0.5
jaxlib>=0.5
jax-autovmap>=0.3.0
numpy>=2.0.2
optax>=0.2.4

[dev]
mypy>=1.0.0
black>=23.0.0
isort>=5.12.0
ruff>=0.0.241
pre-commit>=3.0.0
pytest>=7.0
pytest-xdist>=3.0.0
pytest-cov>=4.0.0
flowjax>=0.3.0
hypothesis[numpy]>=6.90
matplotlib>=3.9.0
nb-clean>=4.0.1

[docs]
sphinx
furo
myst-nb
sphinx-autodoc-typehints
sphinx-copybutton
sphinx-math-dollar
numpydoc
ipykernel
ipywidgets
matplotlib
numpy
