equinox<0.14,>=0.11.0
jax<0.9,>=0.4.36
numpy<2.4,>=1.20.0
