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.