numpy
scipy
jax
jaxopt
numdifftools
patsy
