beartype>=0.21.0
cloudpickle>=3.1.1
diffrax>=0.7.0
einops>=0.8.1
einx>=0.3.0
equinox>=0.12.2
grain>=0.2.11
jax>=0.6.2
jaxtyping>=0.3.2
matplotlib>=3.10.3
ml-collections>=1.1.0
numpy>=2.2.6
optax>=0.2.5
pillow>=11.3.0
tqdm>=4.67.1
