py-torch-cluster (spack): This package consists of a small extension library of highly optimized graph cluster algorithms for the use in PyTorch. 
Keywords: cluster-algorithms|geometric-deep-learning|graph-neural-networks|pytorch