numpy
matplotlib
scipy>=1.7.0
JAX>=0.3.7
