torch
torch_sparse
scikit-learn
torch_geometric
numpy
networkx<2.7
scipy
