transformers
py3nvml
torch
torchvision
