torch
torch_geometric
wandb
scipy
