Torch Geometric

This brings PyTorch Geometric's full graph neural network toolkit to Claude: 60+ GNN layers (GCN, GAT, GraphSAGE, GIN), heterogeneous graph support, and mini-batch training for node, link, and graph-level tasks. The skill covers PyG 2.7+ with clear guidance on the notorious extension wheel installation (pyg-lib, torch-scatter) that trips up newcomers. It explains the critical edge_index COO format and lazy initialization patterns. Use this when you're building GNNs on citation networks, molecular graphs, or point clouds. The MessagePassing API coverage means you can implement custom layers, not just stack built-ins. One thing: PyG's design choices (no built-in activations, manual bidirectional edges) are footguns if you're coming from vanilla PyTorch, and this skill walks through them.

skills.sh installs: 2.1K
GitHub stars: 56.6K
last updated: Aug 21, 2026

View on GitHub

Install

$ npx -y skills add k-dense-ai/scientific-agent-skills --skill torch-geometric --agent claude-code

Installs into .claude/skills of the current project.

Files

PyTorch Geometric (PyG)

PyG is the standard library for Graph Neural Networks built on PyTorch. It provides data structures for graphs, 60+ GNN layer implementations, scalable mini-batch training, and support for heterogeneous graphs.

Installation

Reviewed released torch-geometric 2.8.0.post1 (2026-10-01); CPU examples tested with Python 3.13 / PyTorch 2.14.1. Rolling latest docs identify 2.9.0; check the installed release before adopting new APIs. PyG 2.8 requires PyTorch 2.9+; its original release table covers 2.9–2.12. Our 2.14.1 core tests do not establish every extension/backend combination.

# Install the PyTorch build for your platform from https://pytorch.org/get-started/locally/
uv pip install torch==2.14.1
uv pip install torch-geometric==2.8.0.post1
python -c "import torch, torch_geometric; print(torch.__version__, torch.version.cuda, torch_geometric.__version__)"

SKILL.md on GitHub

Details

Repository: github.com/k-dense-ai/scientific-agent-skills
Categories: AI & Agent Building, Data Science & ML, Marketing & SEO, Go
First seen: Jun 3, 2026