torch-geometric

Build and evaluate graph neural networks with PyG for node classification, link prediction, and heterogeneous graphs.

22|4|Updated May 25, 2026
One-click install
npx skills add https://github.com/crazymsn/academic-skills --skill torch-geometric-crazymsn
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torch-geometric
Source: https://github.com/crazymsn/academic-skills/tree/main/academic-skills/torch-geometric
Command: npx skills add https://github.com/crazymsn/academic-skills --skill torch-geometric-crazymsn

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

PyG provides a practical, scalable framework for designing, training, and evaluating graph neural networks across homogeneous and heterogeneous graphs.

Core Features & Use Cases

  • Extensive GNN ecosystem with 60+ layers, Data and HeteroData structures, and transforms to simplify graph data workflows.
  • Built-in support for homogeneous and heterogeneous graphs, multiple datasets, and scalable training patterns for large graphs.
  • End-to-end workflows for model building, evaluation, explainability, and heterogeneous graph learning.

Quick Start

Install PyG, load a small graph, and run a minimal two-layer GCN to observe graph convolution in action.

Frequently Asked Questions about torch-geometric

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I start with graph neural networks in PyTorch Geometric?

Start with graph neural networks by installing PyG, loading a small graph into a Data object, and running a minimal two-layer GCN to observe graph convolution. This quick start approach demonstrates core concepts and training patterns in action.

How do I build a graph neural network for node classification?

Build a graph neural network for node classification by loading graph data into PyG Data structures, applying a two-layer GCN, and executing standard training patterns to classify nodes. The Skill provides practical examples covering this end-to-end workflow.

Can I process heterogeneous graphs using PyG?

Yes, you can process heterogeneous graphs using PyG HeteroData structures. The framework provides built-in support for heterogeneous graph workflows, allowing you to model complex relationships and run learning tasks across different node and edge types.

What's the best way to handle link prediction with graph data?

Handle link prediction by leveraging PyG's graph data structures and transforms to construct training patterns. The framework supports link prediction workflows alongside node classification and graph classification, providing practical examples for evaluation.

Does PyG support scalable training for large graph datasets?

Yes, PyG supports scalable training patterns for large graph datasets. It provides specialized data loaders and transforms that allow graph neural networks to process and analyze large-scale graph-structured data efficiently.

How do I start with graph neural networks in PyTorch Geometric?

Start with graph neural networks by installing PyG, loading a small graph into a Data object, and running a minimal two-layer GCN to observe graph convolution. This quick start approach demonstrates core concepts and training patterns in action.