torch-geometric
K-Dense-AI/scientific-agent-skills
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.