torch
numpy
Pillow
timm
torchvision
matplotlib