torch
torch_sparse
torch_scatter
sklearn
torch_geometric
numpy
networkx==2.6.3
scipy
