PyTorch Extension Library of Optimized Scatter Operations
https://github.com/rusty1s/pytorch_scatter.git
latest