equinox<0.14,>=0.11.0
jax<0.11,>=0.4.36
numpy<2.5,>=1.20.0
