jax>=0.4.20
optax>=0.1.7
flax>=0.7.5
numpy>=1.23

[dev]
build>=1.2
twine>=5
pytest>=8
pytest-cov>=5
ruff>=0.5
pyright>=1.1
pre-commit>=3
chex>=0.1.8

[docs]
mkdocs>=1.5
mkdocs-material>=9
