torch
pytorch-lightning
torchvision