jax
numpy
scipy
matplotlib
