Skill UI

Reference for building Graph Neural Networks with PyTorch Geometric, covering data objects, datasets, transforms, convolution layers, lazy initialization, and heterogeneous support so you can quickly tackle node, link, and graph tasks.

PyTorch Geometric (PyG) is the definitive standard library for building Graph Neural Networks (GNNs) atop PyTorch. It provides specialized data structures for handling diverse graph types, including heterogeneous graphs. The library offers over 60 GNN layer implementations (such as GCN, GAT, GraphSAGE), full support for dataset loading (Planetoid, TUDataset, OGB), and necessary data transformations. It is essential for deep learning tasks involving node classification, graph classification, and link prediction.