torchdrug

Build custom GNN architectures for molecular and protein science tasks.

13|3|Updated Jun 10, 2026
One-click install
npx skills add https://github.com/tassiovale/claude-code-kit --skill torchdrug-tassiovale
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: torchdrug
Source: https://github.com/tassiovale/claude-code-kit/tree/main/skills/torchdrug
Command: npx skills add https://github.com/tassiovale/claude-code-kit --skill torchdrug-tassiovale

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires torch, torchdrug, torch-scatter, torch-cluster, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill helps users build and deploy custom graph neural network (GNN) architectures for various molecular and protein science tasks, such as drug discovery, protein modeling, and knowledge graph reasoning.

Core Features & Use Cases

  • Custom GNN Architectures: Offers a variety of GNN architectures like GIN, GAT, and SchNet for molecular property prediction, protein modeling, and knowledge graph reasoning.
  • Molecular Property Prediction: Predicts chemical, physical, and biological properties of molecules from their structure.
  • Protein Modeling: Models protein sequences, structures, and properties, including function prediction and structure prediction.
  • Knowledge Graph Reasoning: Predicts missing links and relationships in biological knowledge graphs.
  • Use Case: A researcher uses this Skill to build a custom GNN model for predicting the binding affinity of drug candidates to their target proteins.

Quick Start

Run the following command to install TorchDrug and its dependencies:

pip install torchdrug

Frequently Asked Questions about torchdrug

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

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

You build custom graph neural networks for molecular property prediction using architectures like GIN, GAT, and SchNet. This Skill provides the necessary components to train models that predict chemical, physical, and biological properties directly from molecular structures.

Can I use PyTorch to model protein sequences and predict protein functions?

Yes, you can use PyTorch to model protein sequences, structures, and properties for function prediction. The Skill employs graph neural networks to handle protein modeling tasks, leveraging PyTorch and associated GNN libraries for training.

Does this Skill support knowledge graph reasoning and link prediction?

Yes, this Skill supports knowledge graph reasoning by predicting missing links and relationships in biological knowledge graphs. It applies custom graph neural network architectures to infer connections within complex molecular and biological datasets.

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

The best way to predict drug candidate binding affinity is by building a custom graph neural network model. This Skill enables researchers to design and train GNN architectures specifically for predicting interactions between drug candidates and their target proteins.

Do I need specific GNN libraries installed to run TorchDrug?

Yes, you need PyTorch along with torch-scatter and torch-cluster installed. These GNN-related libraries provide the underlying tensor operations and graph aggregation functions required to build and train the custom molecular and protein models.

Related Skills