jax[cpu]
numpy
scipy
matplotlib
