Zum Hauptinhalt springen
Dieser Inhalt ist noch nicht in Ihrer Sprache verfügbar und wird auf Englisch angezeigt.

Torch Geometric

Skill Verifiziert Aktiv

Guide for building Graph Neural Networks with PyTorch Geometric (PyG). Use this skill whenever the user asks about graph neural networks, GNNs, node classification, link prediction, graph classification, message passing networks, heterogeneous graphs, neighbor sampling, or any task involving torch_geometric / PyG. Also trigger when you see imports from torch_geometric, or the user mentions graph convolutions (GCN, GAT, GraphSAGE, GIN), graph data structures, or working with relational/network data. Even if the user just says 'graph learning' or 'geometric deep learning', use this skill.

Zweck

To serve as a comprehensive guide for building Graph Neural Networks with PyTorch Geometric, enabling users to leverage PyG effectively for various graph learning tasks.

Funktionen

  • Detailed explanation of PyG core concepts (Data, HeteroData, Transforms)
  • Guidance on building GNN models with built-in layers and custom MessagePassing
  • Examples for node classification, graph classification, and link prediction tasks
  • Strategies for scaling GNNs to large graphs using neighbor sampling (NeighborLoader)
  • Comprehensive resources for heterogeneous graph learning and explainability

Anwendungsfälle

  • Learning to build GNNs with PyTorch Geometric from scratch.
  • Implementing node classification, link prediction, or graph classification tasks.
  • Developing GNN models for large graphs that do not fit into GPU memory.
  • Working with heterogeneous graph data structures.

Nicht-Ziele

  • Providing a direct interface to specific GNN models or pre-trained weights.
  • Handling the installation and management of PyTorch or CUDA environments.
  • Covering advanced GNN architectures beyond the scope of PyTorch Geometric's standard offerings.

Workflow

  1. Understand core PyG concepts (Data, HeteroData, Transforms).
  2. Learn to build GNN models using built-in layers or MessagePassing.
  3. Implement task-specific patterns (node/graph classification, link prediction).
  4. Apply scaling strategies (NeighborLoader) for large graphs.
  5. Develop models for heterogeneous graphs and explore explainability.

Praktiken

  • GNN model development
  • Graph data handling
  • Message passing implementation
  • Scalable GNN training
  • Heterogeneous graph learning

Voraussetzungen

  • PyTorch installed
  • PyTorch Geometric (`torch_geometric`) installed via `uv`
  • Optional: `pyg-lib`, `torch-scatter`, `torch-sparse`, `torch-cluster` for accelerated operations

Installation

npx skills add K-Dense-AI/claude-scientific-skills

Führt das Vercel skills CLI (skills.sh) via npx aus — benötigt Node.js lokal und mindestens einen installierten skills-kompatiblen Agent (Claude Code, Cursor, Codex, …). Setzt voraus, dass das Repo dem agentskills.io-Format folgt.

Qualitätspunktzahl

Verifiziert
98 /100
Analysiert 1 day ago

Vertrauenssignale

Letzter Commit3 days ago
Sterne21k
LizenzMIT
Status
Quellcode ansehen

Ähnliche Erweiterungen

PyTorch Lightning

100

Deep learning framework (PyTorch Lightning). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training.

Skill
K-Dense-AI

Nnsight Remote Interpretability

99

Provides guidance for interpreting and manipulating neural network internals using nnsight with optional NDIF remote execution. Use when needing to run interpretability experiments on massive models (70B+) without local GPU resources, or when working with any PyTorch architecture.

Skill
davila7

Pytorch Lightning

99

High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate. Scales from laptop to supercomputer with same code. Use when you want clean training loops with built-in best practices.

Skill
Orchestra-Research

Huggingface Accelerate

99

Simplest distributed training API. 4 lines to add distributed support to any PyTorch script. Unified API for DeepSpeed/FSDP/Megatron/DDP. Automatic device placement, mixed precision (FP16/BF16/FP8). Interactive config, single launch command. HuggingFace ecosystem standard.

Skill
davila7

HuggingFace Accelerate

97

Simplest distributed training API. 4 lines to add distributed support to any PyTorch script. Unified API for DeepSpeed/FSDP/Megatron/DDP. Automatic device placement, mixed precision (FP16/BF16/FP8). Interactive config, single launch command. HuggingFace ecosystem standard.

Skill
Orchestra-Research

TimesFM Forecasting

100

Zero-shot time series forecasting with Google's TimesFM foundation model. Use for any univariate time series (sales, sensors, energy, vitals, weather) without training a custom model. Supports CSV/DataFrame/array inputs with point forecasts and prediction intervals. Includes a preflight system checker script to verify RAM/GPU before first use.

Skill
K-Dense-AI