einops>=0.8.0
equinox>=0.11.5
jax>=0.4.25
jaxlib>=0.4.25
