torch
torch_geometric
networkx
numpy
