equinox >= 0.11.0, != 0.13.3
jax >= 0.5.0, != 0.5.1, != 0.5.2, != 0.6.0, != 0.7.1, != 0.8.2, != 0.9.0
jaxtyping >= 0.2.24
numpy >= 1.20.0
