torch-scatter (pypi): PyTorch Extension Library of Optimized Scatter Operations
Keywords: pytorch|scatter|segment|gather