torch>=2.9.1
torchvision>=0.24.1
numpy>=1.23
Pillow>=10.0.0
PyYAML>=6.0
tqdm>=4.66
dists-pytorch>=0.1.1
timm
triton # or triton-windows

