What problem does it solve?
TorchDrug helps you build and train PyTorch-native graph neural network models for molecules, proteins, and biomedical knowledge graphs, so you can predict properties and perform reasoning tasks without stitching together multiple fragile toolchains.
Core Features & Use Cases
- Molecular property prediction: Train GNNs for classification and regression using curated drug discovery datasets (e.g., BBBP, HIV, Tox21).
- Protein modeling: Apply sequence or structure-aware models (e.g., ESM for sequences, GearNet/SchNet for structures) to tackle function, stability, and localization tasks.
- Knowledge graph reasoning: Perform link prediction on general and biomedical KGs (e.g., FB15k-237, Hetionet) using embedding and reasoning tasks.
- Molecular generation & retrosynthesis: Generate novel molecules and plan synthetic routes via generation and multi-step retrosynthesis workflows.
Quick Start
Ask the AI to propose a TorchDrug workflow to train a GIN-based molecular property predictor on BBBP with AUROC/AUPRC metrics, including dataset loading, model configuration, and a scaffold-split training loop.