torchdrug

Apply graph neural networks to drug discovery tasks with PyTorch.

16|7|Updated Nov 20, 2025
One-click install
npx skills add https://github.com/jackspace/ClaudeSkillz --skill torchdrug
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torchdrug
Source: https://github.com/jackspace/ClaudeSkillz/tree/main/skills/scientific-pkg-torchdrug
Command: npx skills add https://github.com/jackspace/ClaudeSkillz --skill torchdrug

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

TorchDrug provides a comprehensive PyTorch-based toolbox for drug discovery and molecular science, applying graph neural networks to molecules, proteins, and biological knowledge graphs.

Core Features & Use Cases

  • Molecular property prediction, protein modeling, and knowledge-graph reasoning
  • Molecular generation and retrosynthesis planning
  • 40+ datasets and 20+ model architectures for flexible experimentation

Quick Start

Install via pip, load datasets, define models, and train end-to-end using standard PyTorch workflows.

Frequently Asked Questions about torchdrug

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

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

Graph neural networks model molecules as graphs where atoms are nodes and bonds are edges. TorchDrug provides GNN architectures like GIN, GAT, and SchNet integrated with PyTorch, enabling end-to-end molecular property prediction on datasets with RDKit-compatible SMILES strings and protein sequences.

Can I use TorchDrug for protein modeling with AlphaFold and ESM?

Yes. TorchDrug supports protein modeling with compatibility for AlphaFold and ESM protein representations. You can integrate protein sequences and structures into graph-based workflows for knowledge graph reasoning and biomedical applications.

What's the best approach for retrosynthesis planning and molecular generation?

TorchDrug provides graph-based molecular generation and retrosynthesis planning across SMILES strings and reaction graphs. Its 20+ model architectures and 40+ datasets enable flexible experimentation with PyTorch workflows for drug discovery.

How do I work with biomedical knowledge graphs in drug discovery?

TorchDrug integrates molecules, proteins, and biomedical knowledge graphs as unified graph representations. Apply GNN-based reasoning to knowledge graphs for drug discovery tasks like entity linking, relation prediction, and molecular reasoning.

Does TorchDrug handle multiple data formats like SMILES, protein sequences, and structures?

Yes. TorchDrug accepts SMILES strings for molecules, protein sequences and structures for proteins, and reaction graphs. RDKit integration enables seamless conversion and standardization across formats within PyTorch-based workflows.

What are the limitations when scaling TorchDrug to large molecular datasets?

TorchDrug supports 40+ datasets, but graph construction and GNN forward passes scale with molecular size and graph density. Consider memory constraints on GPU when processing large proteins or dense biomedical knowledge graphs in batch training.