jax
jaxopt
numpy
matplotlib
