numpy
matplotlib
scipy>=1.7.0
jaxlib>=0.3.10
JAX>=0.3.10
