jax>=0.4.13
jaxtyping>=0.2.20
typing_extensions>=4.5.0
equinox>=0.11.3
optax>=0.2.0
tqdm>=4.63.2
