py-torch-scatter (spack): This package consists of a small extension library of highly optimized sparse update (scatter and segment) operations for the use in PyTorch, which are missing in the main package. 
Keywords: gather|pytorch|scatter|segment