torchdrug

Train graph neural networks on molecular and biomedical datasets for drug discovery.

15|2|Updated Dec 17, 2025
One-click install
npx skills add https://github.com/rubensliv/k-dense-ai --skill torchdrug-rubensliv
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torchdrug
Source: https://github.com/rubensliv/k-dense-ai/tree/main/scientific-skills/torchdrug
Command: npx skills add https://github.com/rubensliv/k-dense-ai --skill torchdrug-rubensliv

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

TorchDrug enables researchers to build graph-based deep learning workflows for molecules, proteins, and biomedical graphs, accelerating discovery by turning complex biological data into actionable predictions.

Core Features & Use Cases

  • Graph neural networks for molecular property prediction, protein modeling, and knowledge graph reasoning.
  • Rich datasets, models, tasks, and training workflows tailored for chemistry and biology.
  • Seamless integration with RDKit, AlphaFold/ESM, and PyTorch Lightning to streamline experiments.

Quick Start

Train a GIN model on BBBP to predict blood-brain barrier penetration.

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 for molecular property prediction are enabled by training configurable models like GIN on molecular datasets to predict properties such as blood-brain barrier penetration. You can leverage rich dataset pipelines and training workflows tailored for chemistry.

Can I integrate RDKit and PyTorch Lightning for protein modeling workflows?

Yes, protein modeling workflows support seamless integration with RDKit, AlphaFold/ESM, and PyTorch Lightning. This streamlines experiments by combining graph-based deep learning with established structural biology and machine learning frameworks.

What's the best way to build a dataset pipeline for biomedical knowledge graph reasoning?

Building a dataset pipeline for biomedical knowledge graph reasoning utilizes configurable datasets and tasks tailored for biology. The system supports YAML frontmatter for discovery and optional references directories to structure the workflow efficiently.

Does this graph-based deep learning approach work for molecular generation tasks?

Graph-based deep learning applies directly to molecular generation tasks within chemistry workflows. It supports configurable models and training workflows to generate molecular structures based on biological data inputs.

Do I need PyTorch experience to train models for blood-brain barrier penetration?

PyTorch experience is helpful since the system integrates with PyTorch Lightning to streamline training workflows. However, configurable models and YAML-based setups lower the entry barrier for predicting blood-brain barrier penetration.

Why use graph neural networks instead of standard deep learning for drug discovery?

Graph neural networks excel at drug discovery because they directly model molecular, protein, and biomedical graph structures. This captures complex biological relationships that standard deep learning approaches struggle to represent.