PyTorch Extension Library of Optimized Scatter Operations
conda install conda-forge::pytorch_scatter_proc