pytorch_cluster

Graph clustering library

A PyTorch extension library providing optimized graph cluster algorithms

PyTorch Extension Library of Optimized Graph Cluster Algorithms

GitHub

824 stars
15 watching
148 forks
Language: C++
last commit: 7 days 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 787
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,217
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. 875
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 424
prabhuomkar/pytorch-cpp A C++ implementation of PyTorch tutorials 1,965
eduardoleao052/js-pytorch A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. 1,084
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 365
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
benedekrozemberczki/pytorch_geometric_temporal A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models 2,669
graal-research/poutyne A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities 569