torchdrug

Design, train, and evaluate graph-based models for molecular and protein data.

48|6|Updated Mar 9, 2026
One-click install
npx skills add https://github.com/qinyan-ai/qinyan-academic-skills --skill torchdrug-qinyan-ai
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torchdrug
Source: https://github.com/qinyan-ai/qinyan-academic-skills/tree/main/skills/06-%E5%8C%96%E5%AD%A6%E4%BF%A1%E6%81%AF%E4%B8%8E%E8%8D%AF%E7%89%A9%E5%8F%91%E7%8E%B0/torchdrug
Command: npx skills add https://github.com/qinyan-ai/qinyan-academic-skills --skill torchdrug-qinyan-ai

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

TorchDrug provides a unified, modular platform for building, training, and evaluating graph-based models across molecules, proteins, and biomedical knowledge graphs, reducing integration effort and accelerating drug discovery research.

Core Features & Use Cases

  • Graph neural network models for molecular property prediction, protein modeling, and knowledge graph reasoning.
  • Extensive dataset access and standardized task wrappers for rapid prototyping and reproducibility.
  • Interoperability with RDKit/3D structure tools and pre-training approaches for protein sequences and structures.

Quick Start

Install TorchDrug, load a molecular dataset (e.g., BBBP), define a GIN model, and wrap it with 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?

Graph neural networks for molecular property prediction are trained by loading a dataset like BBBP, defining a GIN model, and wrapping it with a PropertyPrediction task to execute training. This modular interface enables rapid prototyping and reproducible experiments.

Can I use this approach for protein modeling and biomedical knowledge graphs?

Yes, the platform supports protein modeling and biomedical knowledge graph reasoning alongside molecular property prediction. It provides pre-training approaches for protein sequences and structures, enabling unified graph-based model design across these biomedical domains.

What's the best way to prototype and compare graph-based architectures for drug discovery?

The best way to prototype and compare graph-based architectures for drug discovery is using a modular platform with standardized task wrappers and extensive dataset access. This allows rapid model design, training, evaluation, and reproducible comparison across different biomedical workflows.

Does this drug discovery workflow integrate with RDKit and 3D structure tools?

Yes, the drug discovery workflow offers interoperability with RDKit and 3D structure tools. This integration supports the processing of molecular graphs and protein structures within the unified model training and evaluation pipeline.

Why use a modular interface for datasets, models, and tasks in biomedical research?

A modular interface for datasets, models, and tasks is used in biomedical research to reduce integration effort and accelerate drug discovery. It enforces standardized wrappers that enable rapid prototyping, architecture comparison, and reproducible experiments across molecules and proteins.