jax
jaxtyping
jax-tqdm
optax
jaxopt
numpy
scipy
matplotlib
pytest

[dev]
