jax[cpu]<=0.4.14,>=0.3.2
numpy<1.25.0,>=1.20.0
scipy<1.11.0,>=1.5.0
