pyGAT
Graph Network Model
An implementation of the Graph Attention Network model using PyTorch.
Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)
3k stars
17 watching
691 forks
Language: Python
last commit: over 1 year ago
Linked from 1 awesome list
attention-mechanismgraph-attention-networksneural-networkspythonpytorchself-attention
Related projects:
Repository | Description | Stars |
---|---|---|
petarv-/gat | An implementation of Graph Attention Networks in TensorFlow for graph representation learning. | 3,233 |
pyg-team/pytorch_geometric | A PyTorch-based library for training and applying Graph Neural Networks to structured data | 21,382 |
tkipf/pygcn | A PyTorch implementation of a graph neural network architecture for semi-supervised classification | 5,198 |
lukemelas/efficientnet-pytorch | A PyTorch implementation of EfficientNet convolutional neural networks | 7,921 |
jwyang/faster-rcnn.pytorch | An object detection implementation built on top of PyTorch, supporting multi-image batch training and multiple GPUs. | 7,693 |
taki0112/ugatit | An unsupervised image-to-image translation model with adaptive normalization and attention mechanisms. | 6,165 |
benedekrozemberczki/appnp | A PyTorch implementation of a graph neural network model that learns personalized node representations | 363 |
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,286 |
timesler/facenet-pytorch | This project provides pre-trained PyTorch models for face detection and recognition tasks. | 4,550 |
szagoruyko/pytorchviz | A tool to visualize PyTorch execution graphs and traces | 3,228 |
benedekrozemberczki/pytorch_geometric_temporal | A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models | 2,677 |
lucidrains/imagen-pytorch | Implements Google's Text-to-Image Neural Network in PyTorch using a cascading DDPM architecture with dynamic clipping and noise level conditioning. | 8,088 |
dawnranger/pytorch-agnn | An implementation of an attention-based graph neural network in PyTorch for semi-supervised learning | 145 |
pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,371 |
benedekrozemberczki/gam | An implementation of a graph classification model using structural attention and PyTorch | 268 |