numpy
jax
scipy
