torch-scatter
torch-sparse
torch-geometric
