numpy
jax
matplotlib
scipy
