torch
torchvision
pytorch-lightning
transformers
