equinox >= 0.11.0, < 0.13
jax >= 0.4.36, < 0.7
numpy >= 1.20.0, < 2.3
