numpy
matplotlib
jax
scipy
