In a (much) earlier post, I described the intuition and some of the math behind a basic graph neural network (GNN) algorithm, GraphSAGE. How can we implement GraphSAGE for an actual task? I’m a PyTorch person and PyG is my go-to for GNN experiments. For much larger graphs, DGL…

This post follows from the intuition and notation in the GraphSAGE post in this series; check it out to follow along better! On the heels of GraphSAGE, Graph Attention Networks (GATs) [1] were proposed with an intuitive extension — incorporate attention into the aggregation and update steps. What are Graph Attention Networks? GATs are an…