pytorch_geometric
Graph neural network library
A PyTorch-based library for training and applying Graph Neural Networks to structured data
Graph Neural Network Library for PyTorch
21k stars
252 watching
4k forks
Language: Python
last commit: 6 days ago
Linked from 6 awesome lists
deep-learninggeometric-deep-learninggraph-convolutional-networksgraph-neural-networkspytorch
josephmisiti/awesome-machine-learning
academic/awesome-datascience
bharathgs/awesome-pytorch-list
krzjoa/awesome-python-data-science
chihming/awesome-network-embedding
szenergy/awesome-lidarRelated projects:
| Repository | Description | Stars |
|---|---|---|
 diego999/pygat | An implementation of the Graph Attention Network model using PyTorch. | 2,914 |
 benedekrozemberczki/pytorch_geometric_temporal | A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models | 2,669 |
 tkipf/pygcn | A PyTorch implementation of a graph neural network architecture for semi-supervised classification | 5,198 |
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 83,959 |
 yunjey/pytorch-tutorial | A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. | 30,251 |
 sherylhyx/pytorch_geometric_signed_directed | A PyTorch Geometric extension library for working with signed and directed graphs | 128 |
 chingyaoc/ggnn.pytorch | An implementation of a neural network architecture for processing graph-structured data and making predictions on nodes. | 465 |
 dsgiitr/d2l-pytorch | An open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch | 4,244 |
 dmlc/dgl | A Python package for building deep learning models on graph data structures | 13,548 |
 bupt-gamma/openhgnn | An open-source toolkit for training and applying heterogeneous graph neural networks using PyTorch and the Deep Graph Library. | 867 |
 facebookresearch/pytorch3d | A deep learning library for 3D data processing and computer vision research using PyTorch | 8,806 |
| pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,428 |
 graphdeeplearning/benchmarking-gnns | A collection of datasets and tools for evaluating graph neural networks | 2,523 |
 szagoruyko/pytorchviz | A tool to visualize PyTorch execution graphs and traces | 3,228 |
 google-deepmind/graph_nets | An open-source library for building graph networks in TensorFlow and Sonnet | 5,360 |