torch
torchvision
transformers[torch]
pillow