jax>=0.9.0
flax>=0.12.3
orbax-checkpoint>=0.11.32
Pillow<12,>=8
numpy>=1.26.0
huggingface-hub>=1.4.1
torch>=2.10.0
torchvision>=0.25.0
einops>=0.8.2

[convert]
torch>=2.10.0
torchvision>=0.25.0
timm>=1.0.25

[cu126]
torch>=2.10.0
torchvision>=0.25.0

[cu128]
torch>=2.10.0
torchvision>=0.25.0

[dev]
pytest>=8.0.0
torch>=2.10.0
torchvision>=0.25.0
timm>=1.0.25
