图药物发现工具

torchdrug 是一个基于图的药物发现工具包，支持分子属性预测（ADMET）、蛋白质建模、知识图推理等功能。它集成了多种 GNN 模型（如 GIN、GAT、SchNet），并提供 40 多个数据集，适用于基于 PyTorch 的生物医学图机器学习。