torch_geometric
torch_scatter
torch_sparse