numpy
jax
jaxopt
scipy
