tdqm
numpy
jax
jaxopt
optax
chex
pytest
matplotlib
build

[dev]
matplotlib
jax
jaxopt
chex
numpy
scipy
tqdm
jax_dataclasses
pytest
black
build
twine
pylint

[doc]
matplotlib
jax
jaxopt
chex
numpy
scipy
tqdm
jax_dataclasses
pytest
black
build
twine
pylint
