alterlab-torchdrug

Featured

Builds PyTorch-native graph neural networks with TorchDrug for molecules and proteins, exposing custom GNN architectures, task/dataset abstractions, molecular generation, retrosynthesis planning, and knowledge-graph reasoning. Use when developing custom graph model layers, predicting protein properties from sequence or structure, or building retrosynthesis and drug-repurposing pipelines; for ready-made featurizers, MoleculeNet benchmarks, and pre-trained models with less code prefer alterlab-deepchem. Part of the AlterLab Academic Skills suite.

AI & Automation 66 stars 13 forks Updated 1 weeks ago MIT

Install

View on GitHub

Quality Score: 89/100

Stars 20%
61
Recency 20%
90
Frontmatter 20%
70
Documentation 15%
100
Issue Health 10%
50
License 10%
100
Description 5%
100

Skill Content

# TorchDrug ## Overview TorchDrug is a comprehensive PyTorch-based machine learning toolbox for drug discovery and molecular science. Apply graph neural networks, pre-trained models, and task definitions to molecules, proteins, and biological knowledge graphs, including molecular property prediction, protein modeling, knowledge graph reasoning, molecular generation, retrosynthesis planning, with 40+ curated datasets and 20+ model architectures. ## When to Use This Skill This skill should be used when working with: **Data Types:** - SMILES strings or molecular structures - Protein sequences or 3D structures (PDB files) - Chemical reactions and retrosynthesis - Biomedical knowledge graphs - Drug discovery datasets **Tasks:** - Predicting molecular properties (solubility, toxicity, activity) - Protein function or structure prediction - Drug-target binding prediction - Generating new molecular structures - Planning chemical synthesis routes - Link prediction in biomedical knowledge bases - Training graph neural networks on scientific data **Libraries and Integration:** - TorchDrug is the primary library - Often used with RDKit for cheminformatics - Compatible with PyTorch and PyTorch Lightning - Integrates with AlphaFold and ESM for proteins ## Getting Started ### Installation ```bash # TorchDrug 0.2.1 (last release, Jul 2023) requires Python >=3.7,<3.11 and # torch >=1.8. It will NOT solve on Python 3.11+ — pin an older interpreter: uv venv --python 3.10 uv pip install...

Details

Author
AlterLab-IEU
Repository
AlterLab-IEU/AlterLab-Academic-Skills
Created
5 months ago
Last Updated
1 weeks ago
Language
Python
License
MIT

Integrates with

Bundled in these plugins

Similar Skills

Semantically similar based on skill content — not just same category

AI & Automation Featured

alterlab-deepchem

Runs molecular machine learning with DeepChem — diverse featurizers, pre-built MoleculeNet benchmark datasets, and pre-trained models (ChemBERTa, GROVER) for property prediction (ADMET, toxicity, solubility) via traditional ML or graph neural networks. Use when running end-to-end molecular ML experiments that need MoleculeNet benchmarks, scaffold splitting, or ready-made models with minimal setup; for building custom PyTorch graph architectures prefer alterlab-torchdrug, and for standalone molecule-to-feature-vector generation prefer alterlab-molfeat. Part of the AlterLab Academic Skills suite.

66 Updated 1 weeks ago
AlterLab-IEU
AI & Automation Featured

alterlab-pytdc

Loads Therapeutics Data Commons (TDC, PyTDC) AI-ready drug-discovery datasets and benchmarks — ADME, toxicity, drug-target interaction (DTI), scaffold splits, and molecular oracles for therapeutic ML and pharmacological prediction. Use when fetching a standardized benchmark dataset, applying scaffold or cold-split evaluation, or sourcing labeled molecules for ADMET, toxicity, or DTI modeling. Sources data, splits, and oracles only — defer molecular featurization (ECFP/fingerprints), model training, and transformers to a molecular-ML skill (e.g. deepchem). Part of the AlterLab Academic Skills suite.

66 Updated 1 weeks ago
AlterLab-IEU
AI & Automation Featured

deepchem

Molecular ML with diverse featurizers and pre-built datasets. Use for property prediction (ADMET, toxicity) with traditional ML or GNNs when you want extensive featurization options and MoleculeNet benchmarks. Best for quick experiments with pre-trained models, diverse molecular representations. For graph-first PyTorch workflows use torchdrug; for benchmark datasets use pytdc.

44,401 Updated today
K-Dense-AI