pytorch 
torchvision
transformers