pytorch_cluster

Graph clustering library

A PyTorch extension library providing optimized graph cluster algorithms

PyTorch Extension Library of Optimized Graph Cluster Algorithms

GitHub

838 stars
15 watching
151 forks
Language: C++
last commit: 3 months ago
cluster-algorithmsgeometric-deep-learninggraph-neural-networkspytorch

Related projects:

Repository Description Stars
benedekrozemberczki/clustergcn A PyTorch implementation of a clustering algorithm for graph neural networks 788
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
speedinghzl/pytorch-segmentation-toolbox Provides PyTorch implementations of deep learning models for semantic segmentation 772
shahsohil/dcc An implementation of Deep Continuous Clustering algorithms using PyTorch. 209
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,236
benedekrozemberczki/splitter A PyTorch implementation of node representation learning using multiple social contexts 213
jacobgil/pytorch-pruning This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. 877
kaixhin/grokking-pytorch An introduction to using PyTorch for deep learning tasks 1,194
percyliang/brown-cluster A C++ implementation of the Brown word clustering algorithm for hierarchical word grouping 425
prabhuomkar/pytorch-cpp A C++ implementation of PyTorch tutorials 1,978
eduardoleao052/js-pytorch A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. 1,093
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 366
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,044
benedekrozemberczki/pytorch_geometric_temporal A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models 2,694
graal-research/poutyne A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities 572