torch-geometric

SKILL.md

PyTorch Geometric is a library built on PyTorch for developing and training Graph Neural Networks (GNNs). Apply this skill for deep learning on graphs and irregular structures, including mini-batch processing, multi-GPU training, and geometric deep learning applications.

PyG represents graphs using the torchgeometric.data.Data class with these key attributes:

Important: These attributes are not mandatory—extend Data objects with custom attributes as needed.

Graph Neural Networks (PyG). Node/graph classification, link prediction, GCN, GAT, GraphSAGE, heterogeneous graphs, molecular property prediction, for geometric deep learning. Source: jackspace/claudeskillz.

View raw