numpy
scipy
jax
jaxopt
numdifftools
