numpy>=1.22.0
matplotlib
jax
scipy
