torch
torchvision
pytorch_lightning
