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)

GitHub

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

Backlinks from these awesome lists:

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,251
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,669
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 83,959
benedekrozemberczki/gam An implementation of a graph classification model using structural attention and PyTorch 268