torchvision
torch
transformers
numpy

