torch
Pillow
torchvision
timm
cupy