torch
torchvision
timm
pillow
safetensors