GitHub repository: yongbinfeng/pytorch_sparse
Description: PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations