equinox>=0.11
jax<0.8,>=0.7.0
jaxlib<0.8,>=0.7.0
