pytorch_cluster

PyTorch Extension Library of Optimized Graph Cluster Algorithms

MIT License

Downloads
14K
Stars
752
View Code on GitHub View on X
Ecosystems: PyTorch

Bot releases are hidden (Show)

pytorch_cluster - 1.2.3

Published by rusty1s over 5 years ago

pytorch_cluster - 1.2.2

Published by rusty1s almost 6 years ago

  • Iterative Farthest Point Sampling
  • kNN and Radius Graph
  • Nearest
pytorch_cluster - 1.1.5

Published by rusty1s about 6 years ago

  • Voxel Grid CUDA Bugfix
pytorch_cluster - 1.1.4

Published by rusty1s about 6 years ago

  • now uses new extension API
pytorch_cluster - 1.1.3

Published by rusty1s about 6 years ago

PyTorch 0.4.1 adjustments

pytorch_cluster - 1.1.2

Published by rusty1s over 6 years ago

  • Graclus GPU OOM Fixes
pytorch_cluster - 1.1.1

Published by rusty1s over 6 years ago

PyTorch 0.4.0

pytorch_cluster - 1.0.3

Published by rusty1s over 6 years ago

Graclus GPU Bugfixes:

  • self-loops are no longer a problem
  • edge weights of zero are handled correctly
pytorch_cluster - 1.0.1

Published by rusty1s over 6 years ago

  • huge rebuild
  • added graclus gpu and cpu implementation
pytorch_cluster - 0.2.4

Published by rusty1s over 6 years ago

  • huge performance boost on GPU
pytorch_cluster - 0.2.3

Published by rusty1s over 6 years ago

Added dense grid cluster implementation

pytorch_cluster - 0.1.1

Published by rusty1s over 6 years ago

Package Rankings
Top 19.12% on Conda-forge.org
Top 11.85% on Spack.io
Top 1.89% on Pypi.org
Badges
Extracted from project README
PyPI Version Testing Status Linting Status Code Coverage
Related Projects
pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

28 Jul 2018 939
torchdrug

A powerful and flexible machine learning platform for drug discovery

10 Aug 2021 1,427
DIGRAC_Directed_Clustering

Official code for the LoG2022 paper -- DIGRAC: Digraph Clustering Based on Flow Imbalance.

25 Jan 2022 5
torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

12 Aug 2019 568
pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations

16 Dec 2017 1,460
torchview

torchview: visualize pytorch models

08 Nov 2022 803
pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch

27 Dec 2017 165
nvdiffrec

Official code for the CVPR 2022 (oral) paper "Extracting Triangular 3D Models, Materials, and Lig...

24 Nov 2021 2,065
pytorch_geometric_signed_directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for...

28 Oct 2021 116
PyTorchTricks

Some tricks of pytorch...

23 Dec 2019 1,153
graphlearn-for-pytorch

A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training an...

28 Mar 2023 113
torchrec

Pytorch domain library for recommendation systems

12 Jul 2021 1,889
Pointcept

Pointcept: a codebase for point cloud perception research. Latest works: PTv3 (CVPR'24 Oral), PPT...

21 Mar 2023 1,143
torch-points3d

Pytorch framework for doing deep learning on point clouds.

06 Dec 2019 2,463
pytorch-styleguide

An unofficial styleguide and best practices summary for PyTorch

14 Apr 2019 1,909