torch-geometric

Build, train, and deploy graph neural networks with PyTorch Geometric.

1|Updated Mar 4, 2026
One-click install
npx skills add https://github.com/Hung-3008/agusta --skill torch-geometric-hung-3008
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torch-geometric
Source: https://github.com/Hung-3008/agusta/tree/main/.agents/skills/torch-geometric
Command: npx skills add https://github.com/Hung-3008/agusta --skill torch-geometric-hung-3008

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires torch, torch-geometric, numpy, and includes scripts (resource) and references (resource) components.

What problem does it solve?

PyTorch Geometric (PyG) enables building, training, and deploying graph neural networks with a rich set of built-in layers, datasets, and tooling.

Core Features & Use Cases

  • Extensive collection of 40+ graph neural network layers and prebuilt architectures (GCN, GAT, GraphSAGE, GIN, and more) for node and graph tasks.
  • Integrated datasets catalog, transforms, and utilities to train on benchmark datasets and real-world graphs, including homogeneous and heterogeneous graphs, molecular data, and large-scale networks.
  • Quick-start templates and reference materials to accelerate development, experimentation, and deployment of graph-based models.

Quick Start

Install PyG and run a simple GCN on Cora to observe node predictions.

Frequently Asked Questions about torch-geometric

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

FAQPage Schema
How do I build graph neural networks for node classification using PyTorch?

Build graph neural networks for node classification using PyTorch Geometric (PyG) by leveraging quick-start templates and architectures like GCN and GraphSAGE. PyG provides prebuilt layers and integrated datasets to train models and generate node predictions on benchmark graphs.

What graph neural network architectures are available for graph classification tasks?

Graph classification tasks utilize over 40 prebuilt graph neural network layers in PyG, including GCN, GAT, GraphSAGE, and GIN. These architectures support modular model construction to train and deploy models across molecular data and large-scale networks.

Can I train models on heterogeneous graphs with PyTorch Geometric?

Train models on heterogeneous graphs using PyTorch Geometric, which includes integrated datasets and utilities specifically designed for both homogeneous and heterogeneous graphs. Transform pipelines and modular model construction facilitate training across diverse graph structures.

Does PyG support link prediction and molecular property prediction?

PyG supports link prediction and molecular property prediction through its extensive collection of neural network layers and integrated datasets catalog. Researchers can apply transforms and utilize templates to accelerate experimentation across these graph-based tasks.

What do I need to install to start building models with PyTorch Geometric?

Install PyTorch Geometric, PyTorch, and NumPy to start building graph neural networks. PyG provides quick-start templates and reference materials to run a simple GCN on the Cora dataset, enabling immediate observation of node predictions and accelerating development.