jax>=0.4.11
jaxtyping>=0.2.20
equinox>=0.11.0
