TorchDrug

TorchDrug 是专为药物发现设计的机器学习平台，它融合了图机器学习（包括图神经网络、几何深度学习和知识图谱）、深度生成模型以及强化学习等多种技术。该平台提供了全面而灵活的接口，支持在 PyTorch 中快速构建药物发现模型原型。