torch
torchvision
timm
albumentations