torchdrug

Create and apply PyTorch-native Graph Neural Networks for molecular and protein data analysis.

1|Updated Jan 14, 2026
One-click install
npx skills add https://github.com/Sologa/codex-pipeline --skill torchdrug-sologa
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torchdrug
Source: https://github.com/Sologa/codex-pipeline/tree/main/.codex/skills/torchdrug
Command: npx skills add https://github.com/Sologa/codex-pipeline --skill torchdrug-sologa

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

This Skill simplifies the development and application of Graph Neural Networks (GNNs) for complex scientific domains like drug discovery, molecular modeling, and protein analysis, abstracting away low-level implementation details.

Core Features & Use Cases

  • GNN Development: Build custom GNN architectures for molecules, proteins, and knowledge graphs.
  • Task-Specific Tools: Utilize pre-defined tasks for property prediction, generation, and reasoning.
  • Data Handling: Load and process diverse scientific datasets (molecules, proteins, KGs).
  • Use Case: Predict the binding affinity of novel drug candidates to a target protein using a GNN model trained on existing protein-ligand interaction data.

Quick Start

Use the torchdrug skill to predict molecular properties using the BBBP dataset.

Frequently Asked Questions about torchdrug

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

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

To build a graph neural network for molecular property prediction, you can use this Skill to create PyTorch-native GNN architectures. It provides pre-defined tasks and data loaders for molecular datasets like BBBP to streamline the training process.

Can I use PyTorch Lightning for training GNN models on protein data?

Yes, you can use PyTorch Lightning for training GNN models on protein data. This Skill supports integration with scientific computing frameworks like PyTorch Lightning to handle the training of protein modeling and analysis architectures.

How does graph neural network modeling work for drug discovery tasks?

Graph neural network modeling for drug discovery works by representing molecules and proteins as graphs to capture structural relationships. This Skill applies GNNs to these representations to perform tasks like molecular generation and knowledge graph reasoning.

Does this GNN framework support integration with RDKit for molecular data handling?

Yes, this GNN framework supports integration with RDKit for molecular data handling. It facilitates the loading and processing of diverse scientific datasets by combining PyTorch-native GNNs with cheminformatics libraries.

What is the best way to predict drug candidate binding affinity to a target protein?

The best way to predict binding affinity is to train a GNN model on existing protein-ligand interaction data. This Skill enables you to build custom architectures that predict how novel drug candidates bind to target proteins.

Can I perform knowledge graph reasoning for scientific datasets using PyTorch?

Yes, you can perform knowledge graph reasoning using PyTorch. This Skill enables the creation of custom Graph Neural Networks specifically designed to load diverse scientific datasets and execute reasoning tasks.