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