torch
timm
torchvision