torch
torchvision
albumentations
timm
