equinox >= 0.11.0, < 0.14
jax >= 0.4.36, < 0.9
numpy >= 1.20.0, < 2.4
