numpy
jax
jaxopt
chex
pytest
matplotlib
build