torch
torchvision
safetensors
timm