equinox >= 0.11.0, < 0.14
jax >= 0.4.30, < 0.11
jaxtyping >= 0.2.24, < 0.4
lineax >= 0.0.5, <= 0.1.0
numpy >= 1.20.0, < 2.5
