equinox<=0.11.11,>=0.11.0
jax<=0.5.0,>=0.3.2
numpy<=2.2.2,>=1.20.0
