torch
torchvision
torchaudio
tqdm
torch-geometric
transformers
torch-scatter