torchdrug

Train graph-based models for chemistry and protein tasks with TorchDrug.

1|2|Updated Apr 29, 2026
One-click install
npx skills add https://github.com/fuzzy-dynamics/strings --skill torchdrug-fuzzy-dynamics
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torchdrug
Source: https://github.com/fuzzy-dynamics/strings/tree/main/packages/skills/torchdrug
Command: npx skills add https://github.com/fuzzy-dynamics/strings --skill torchdrug-fuzzy-dynamics

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

TorchDrug provides a unified, PyTorch-based toolkit enabling researchers to train and deploy graph-based ML models for chemistry and protein tasks, reducing the complexity of building end-to-end pipelines.

Core Features & Use Cases

  • Molecular property prediction: predict properties for drug discovery datasets (BBBP, HIV, QM9, etc.).
  • Protein modeling: analyze sequences and structures with graph-based encoders like GearNet, ESM, ProteinBERT.
  • Knowledge graph reasoning: perform link prediction on biomedical graphs (Hetionet, FB15k-237).
  • Molecular generation & retrosynthesis: generate novel molecules and plan synthetic routes.
  • Datasets & tooling: access 40+ datasets and integrate with RDKit, AlphaFold, and other tools.

Quick Start

Load a dataset, choose a GNN backbone like GIN, and wrap it in a PropertyPrediction task to start training.

Frequently Asked Questions about torchdrug

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

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

To perform molecular property prediction, load a dataset like BBBP, select a GNN backbone such as GIN, and wrap it in a PropertyPrediction task to start training your graph neural network.

What is the best way to build a unified pipeline for protein modeling and retrosynthesis?

Protein modeling and retrosynthesis workflows can be built in a single integrated PyTorch toolkit, utilizing graph-based encoders like GearNet and modular components for synthetic route planning.

Can I use PyTorch Lightning for scalable training of knowledge graph reasoning models?

Yes, knowledge graph reasoning models for datasets like Hetionet integrate directly with PyTorch Lightning, enabling scalable training for link prediction on biomedical graphs.

Does TorchDrug support 3D geometry for protein structure analysis?

Yes, TorchDrug supports 3D geometry processing with multiple graph neural networks, allowing researchers to analyze protein sequences and structures using encoders like GearNet and ProteinBERT.

What datasets are available for drug discovery model training?

The toolkit provides access to over 40 datasets for drug discovery, including BBBP, HIV, QM9, EnzymeCommission, and Hetionet, covering molecular property prediction and knowledge graph reasoning tasks.