numpy
jax
jaxlib

[dev]
pytest
pytest-cov
optax

[docs]
sphinx
myst-parser
furo
sphinx-copybutton
matplotlib

[gen]
classy

[train]
optax
