jax
matplotlib
numpy
scipy
