torch-geometric

Build and benchmark Graph Neural Networks with PyTorch Geometric.

94|11|Updated Mar 26, 2026
One-click install
npx skills add https://github.com/swaruplab/operon --skill torch-geometric-swaruplab
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torch-geometric
Source: https://github.com/swaruplab/operon/tree/main/src-tauri/protocols/torch-geometric
Command: npx skills add https://github.com/swaruplab/operon --skill torch-geometric-swaruplab

SYSTEM DOCUMENTATION & REQUIREMENTS

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

What problem does it solve?

Graph Neural Network development and benchmarking on PyTorch Geometric, enabling researchers and engineers to implement, train, and evaluate node/graph classification, link prediction, and molecular-property tasks with practical guidance.

Core Features & Use Cases

  • Supports GCN, GAT, GraphSAGE, GIN, and other PyG layers for diverse graph architectures.
  • Provides templates, benchmarks, and examples for node classification, graph classification, and heterogeneous graphs.
  • Use cases include social networks, citation networks, molecular property prediction, and large-scale graph benchmarks.

Quick Start

Install PyTorch and PyG, load a small dataset like Cora, and run a minimal Graph Neural Network example to see results.

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 Geometric?

To build Graph Neural Networks for node classification, this Skill provides ready-to-use templates and benchmark scripts using PyTorch Geometric. You can implement and evaluate models like GCN and GraphSAGE on standard citation datasets such as Cora.

Can I implement link prediction and graph classification with PyG layers?

Yes, you can implement link prediction and graph classification using PyG layers. It supplies model templates and examples for diverse graph architectures including GAT and GIN across social and molecular domains.

What's the best way to benchmark molecular property prediction tasks in PyTorch?

The best way to benchmark molecular property prediction tasks is using the provided benchmark scripts and PyG datasets. These templates allow researchers to efficiently train and evaluate graph neural networks on molecular data.

Do I need PyTorch and PyG installed to run heterogeneous graph benchmarks?

Yes, you need PyTorch, PyG, and NumPy installed as dependencies to run heterogeneous graph benchmarks. These frameworks provide the foundational tensor computations and graph structures required by the templates.

Are there ready-to-use templates for social network graph analysis?

Yes, there are ready-to-use templates for social network graph analysis. The Skill includes examples covering social networks and citation networks to help developers quickly deploy graph neural network architectures.