Extension library of highly optimized sparse update (scatter and segment) operations
conda install ab-geo::torch-scatter