TorchDrug 是基于 PyTorch 的药物发现深度学习库，覆盖分子性质、生成、DTI、知识图谱推理等任务。
安装与部署
pip install torchdrug（需匹配的 PyTorch）；建议在 GPU 环境使用。
典型用法
- 用内置数据集与模型快速搭实验
- 做分子生成、性质预测、知识图谱推理
- 复用预置的图模型组件
- 作为研究新方法的脚手架
上手提示
适合需要在统一框架下试多种任务的研究者；生产部署前评估其维护活跃度与版本兼容。
延伸资源
- 对照 176《PyTorch Geometric》、177《DGL-LifeSci》。