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