torch
torchvision
timm
pytorch-lightning
