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