PyTorch Geometric (PyG)

PyTorch Geometric (PyG) is a powerful library built on PyTorch that simplifies the implementation of Graph Neural Networks. This skill enables Claude to assist with geometric deep learning tasks, including node and graph classification, link prediction, and heterogeneous graph processing. It provides standardized patterns for implementing state-of-the-art architectures like GCN, GAT, and GraphSAGE, while offering best practices for efficient mini-batching, large-scale graph sampling, and custom message-passing layer development.

Key Features

Use Cases