torch
torchvision 
transformers