torch>=2.1
torchvision
timm
Pillow