torch
torch-geometric
numpy
tqdm
