torchdrug

Build, train, and deploy graph neural network workflows for molecules and proteins.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

TorchDrug provides a unified platform to design, train, and deploy graph neural networks for chemistry and biology, enabling researchers to build advanced models without assembling disparate tools.

Core Features & Use Cases

  • Graph neural network architectures for molecules and proteins (GIN, GCN, GAT, GearNet) with 3D support
  • Knowledge graph reasoning and retrosynthesis workflows
  • Data handling, pre-training, and transfer learning with datasets (BBBP, Hetionet, QM9, ZINC, etc.)
  • End-to-end training pipelines, property prediction, generation, and retrosynthesis planning

Quick Start

Load a molecular dataset, initialize a GIN model, wrap it in a PropertyPrediction task, and train with a scaffold split.

Frequently Asked Questions about torchdrug

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

FAQPage Schema
How do I build graph neural networks for molecular property prediction?

To perform molecular property prediction, load a dataset like BBBP, initialize a GNN architecture such as GIN or GCN, wrap it in a PropertyPrediction task, and train the model using an end-to-end pipeline with a scaffold split.

What graph neural network architectures are available for protein modeling?

Protein modeling is supported by multiple graph neural network architectures, including GIN, GCN, GAT, and GearNet, with specific extensions available for handling 3D molecular structures.

Can I use knowledge graph reasoning for drug discovery workflows?

Yes, knowledge graph reasoning is fully supported for drug discovery workflows, allowing researchers to perform complex biomedical reasoning using datasets like Hetionet directly within the training pipeline.

What's the best way to plan retrosynthesis using graph neural networks?

Retrosynthesis planning is executed through dedicated workflows that leverage graph neural networks, providing modular components for data handling and model training to support end-to-end chemical reaction prediction.

Does this platform support transfer learning for biomedical datasets?

Yes, transfer learning is supported across 40+ datasets like QM9 and ZINC, enabling researchers to pre-train models on specific tasks and transfer learned representations to new biomedical predictions.