numpy
scikit-learn
matplotlib

[all]
numpy
scikit-learn
matplotlib
black
black[jupyter]
pip
bump2version
wheel
watchdog
flake8
tox
coverage
Sphinx
furo
sphinx-copybutton
twine
autopep8
pytest
jax>=0.4.23
jaxlib>=0.4.23
chex>=0.1.85
optax>=0.2.0

[dev]
black
black[jupyter]
pip
bump2version
wheel
watchdog
flake8
tox
coverage
Sphinx
furo
sphinx-copybutton
twine
autopep8
pytest

[docs]
Sphinx
furo
sphinx-copybutton

[export]
flatbuffers>=25.12.19

[jax]
jax>=0.4.23
jaxlib>=0.4.23
chex>=0.1.85
optax>=0.2.0

[jax-cuda12]
jax[cuda12]>=0.4.23
chex>=0.1.85
optax>=0.2.0
