numpy>=1.20.0
scipy>=1.7.0
jax[cpu]>=0.4.0
