torch
Pillow
torchvision
timm