pytorch_geometric
by pyg-team
Graph Neural Network Library for PyTorch
AI summary
Graph neural network library
A PyTorch-based library for training and applying Graph Neural Networks to structured data
- stars
- 21.6K
- forks
- 3.7K
- watching
- 254
- awesome lists
- 6
Similar projects
Found by comparing what the projects do, not just their names.
diego999/pygat2.9K
Graph Network Model
An implementation of the Graph Attention Network model using PyTorch.
Temporal graph NN library
A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models
tkipf/pygcn5.2K
GCN model
A PyTorch implementation of a graph neural network architecture for semi-supervised classification
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
Deep Learning Tutorials
A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage.
Graph neural network library
A PyTorch Geometric extension library for working with signed and directed graphs
Graph neural network
An implementation of a neural network architecture for processing graph-structured data and making predictions on nodes.
Deep learning book adaptation
An open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch
dmlc/dgl13.6K
Graph DL framework
A Python package for building deep learning models on graph data structures
Graph Neural Network Toolkit
An open-source toolkit for training and applying heterogeneous graph neural networks using PyTorch and the Deep Graph Library.
3D data processor
A deep learning library for 3D data processing and computer vision research using PyTorch
pytorch/examples22.5K
PyTorch examples
A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning.
Graph network benchmarking suite
A collection of datasets and tools for evaluating graph neural networks
Graph visualizer
A tool to visualize PyTorch execution graphs and traces
Graph network builder
Builds graph networks in TensorFlow using a library provided by DeepMind's AI research organization.
Featured in 6 awesome lists
Each link jumps to the spot where the list mentions pytorch_geometric.
