← all repositories

rusty1s/pytorch_cluster

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

926 stars C++ ML Frameworks
pytorch_cluster
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.