torch
torchvision
torchaudio
tqdm
torch-geometric
transformers