numpy
jax
jaxopt
matplotlib
