Image
torch
torchvision
pytorch-lightning
timm
