A PyTorch Implementation of "Watch Your Step: Learning Node Embeddings via Graph Attention" (NeurIPS 2018).
GPL-3.0 License
BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)
A PyTorch implementation of "Graph Classification Using Structural Attention" (KDD 2018).
A PyTorch implementation of "Capsule Graph Neural Network" (ICLR 2019).
A PyTorch implementation of the Transformer model in "Attention is All You Need".
A recurrent attention module consisting of an LSTM cell which can query its own past cell states ...
A scalable Gensim implementation of "Learning Role-based Graph Embeddings" (IJCAI 2018).
PyTorch Implementation and Explanation of Graph Representation Learning papers: DeepWalk, GCN, Gr...
A PyTorch implementation of "Graph Wavelet Neural Network" (ICLR 2019)
A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computat...
CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
My implementation of the original GAT paper (Veličković et al.). I've additionally included the p...
PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models ...
Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://a...