torch
transformers
torchvision
numpy