torch
torchvision
timm