torch>=2.0 torch_geometric>=2.4 torch_sparse torch_scatter numpy scipy scikit-learn matplotlib