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