PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
conda install pyg::pytorch-sparse