numpy
scipy
jax
