tdqm
numpy
jax
jaxopt
optax
chex
pytest
matplotlib
build