torch
Pillow
torchvision
matplotlib
transformers