rusty1s/pytorch_cluster
A PyTorch C++ extension library of optimized graph clustering algorithms for deep learning.

Velocity · 7d
+0.3
★ / day
Trend
→steady
star history
This library provides highly optimized implementations of graph cluster algorithms designed to integrate with PyTorch. It includes algorithms such as Graclus for weighted graph cuts, Voxel Grid pooling, Iterative Farthest Point Sampling, k-NN and radius graph generation, and Random Walk sampling. These are commonly used as building blocks in graph neural networks, point cloud processing (e.g., PointNet++), and other geometric deep learning architectures.