equinox >= 0.11.0, < 0.14
jax >= 0.4.30, < 0.9
jaxtyping >= 0.2.24, < 0.4.0
lineax >= 0.0.5, <= 0.0.8
numpy >= 1.20.0, < 2.4
