numpy
matplotlib
scipy>=1.8.0
JAX>=0.3.7
