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
22k stars
254 watching
4k forks
Language: Python
last commit: 2 months 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,935 |
 benedekrozemberczki/pytorch_geometric_temporal | A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models | 2,694 |
 tkipf/pygcn | A PyTorch implementation of a graph neural network architecture for semi-supervised classification | 5,214 |
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,978 |
 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,401 |
 sherylhyx/pytorch_geometric_signed_directed | A PyTorch Geometric extension library for working with signed and directed graphs | 131 |
 chingyaoc/ggnn.pytorch | An implementation of a neural network architecture for processing graph-structured data and making predictions on nodes. | 466 |
 dsgiitr/d2l-pytorch | An open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch | 4,256 |
 dmlc/dgl | A Python package for building deep learning models on graph data structures | 13,601 |
 bupt-gamma/openhgnn | An open-source toolkit for training and applying heterogeneous graph neural networks using PyTorch and the Deep Graph Library. | 879 |
 facebookresearch/pytorch3d | A deep learning library for 3D data processing and computer vision research using PyTorch | 8,889 |
| pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,530 |
 graphdeeplearning/benchmarking-gnns | A collection of datasets and tools for evaluating graph neural networks | 2,537 |
 szagoruyko/pytorchviz | A tool to visualize PyTorch execution graphs and traces | 3,251 |
 google-deepmind/graph_nets | Builds graph networks in TensorFlow using a library provided by DeepMind's AI research organization. | 5,370 |