torchdrug

Train PyTorch graph neural networks for molecular and protein property prediction.

4|Updated Mar 2, 2026
One-click install
npx skills add https://github.com/shushuzn/Rairos --skill torchdrug-shushuzn
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: torchdrug
Source: https://github.com/shushuzn/Rairos/tree/main/skills/torchdrug
Command: npx skills add https://github.com/shushuzn/Rairos --skill torchdrug-shushuzn

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

TorchDrug helps you build and train PyTorch-native graph neural network models for molecules, proteins, and biomedical knowledge graphs, so you can predict properties and perform reasoning tasks without stitching together multiple fragile toolchains.

Core Features & Use Cases

  • Molecular property prediction: Train GNNs for classification and regression using curated drug discovery datasets (e.g., BBBP, HIV, Tox21).
  • Protein modeling: Apply sequence or structure-aware models (e.g., ESM for sequences, GearNet/SchNet for structures) to tackle function, stability, and localization tasks.
  • Knowledge graph reasoning: Perform link prediction on general and biomedical KGs (e.g., FB15k-237, Hetionet) using embedding and reasoning tasks.
  • Molecular generation & retrosynthesis: Generate novel molecules and plan synthetic routes via generation and multi-step retrosynthesis workflows.

Quick Start

Ask the AI to propose a TorchDrug workflow to train a GIN-based molecular property predictor on BBBP with AUROC/AUPRC metrics, including dataset loading, model configuration, and a scaffold-split training loop.

Frequently Asked Questions about torchdrug

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

FAQPage Schema
How do I train a graph neural network for molecular property prediction in PyTorch?▼

Train graph neural networks for molecular property prediction by defining a YAML workflow that loads curated datasets like BBBP or Tox21, configures a GNN model, and runs a scaffold-split training loop with AUROC and AUPRC metrics.

Can I use PyTorch to model protein structures and sequences for function prediction?▼

Model protein structures and sequences for function prediction by applying PyTorch-native sequence models like ESM or structure-aware models like GearNet and SchNet to tackle stability, localization, and function tasks.

What's the best way to perform link prediction on biomedical knowledge graphs?▼

Perform link prediction on biomedical knowledge graphs like Hetionet or FB15k-237 by using PyTorch-based embedding and reasoning tasks to complete missing relationships within the graph structure.

Does PyTorch support molecular generation and retrosynthesis planning workflows?▼

PyTorch supports molecular generation and retrosynthesis planning through workflows that generate novel molecules and plan synthetic routes using multi-step retrosynthesis tasks defined via YAML metadata.

How do I set up dataset splitting and evaluation metrics for graph neural networks?▼

Set up dataset splitting and evaluation metrics for graph neural networks by configuring YAML-defined metadata that specifies task, criterion, and metrics setup alongside proper dataset splitting for training workflows.

Why use a unified PyTorch toolchain for graph neural networks instead of stitching multiple frameworks together?▼

Use a unified PyTorch toolchain for graph neural networks to predict properties and perform reasoning tasks without stitching together multiple fragile toolchains for molecules, proteins, and biomedical knowledge graphs.