torch
torch_geometric
numpy
matplotlib
scikit-learn
pyyaml
wandb
tabulate
scipy
