PyTorch Extension Library of Optimized Scatter Operations - View it on GitHub