--index-url https://download.pytorch.org/whl/cu128
--extra-index-url https://pypi.org/simple

torch>=2.2,<2.4
torchvision>=0.17,<0.19
timm>=0.9,<1.0
numpy>=1.24,<2.0
tqdm>=4.66
pyyaml>=6.0
einops>=0.7
