numpy>=1.22.0
matplotlib
jax>=0.4.24
scipy
