jax>=0.6.0
tqdm>=4.67.1
h5py>=3.13.0
ruamel.yaml>=0.18.10
Pillow>=10.0.0
robo-goggles>=0.1.7
numpy>=2.2.6
gdown>=5.2.0
grain>=0.2.13
orbax-checkpoint>=0.11.31

[cuda12]
jax[cuda12]>=0.6.0

[cuda13]
jax[cuda13]>=0.10.1

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
pre_commit==4.0.1
matplotlib>=3.10.1
ruff>=0.9.0
basedpyright
pydoclint>=0.8.3

[download]
gdown>=5.1.0
scipy>=1.10.0
tensorflow-cpu>=2.15.0

[hf]
huggingface_hub>=1.0
hf_transfer>=0.1.6
hf-xet>=0.1.0
