昌平实验室联合华为发布TorchX模型体系

IT时代网9月21日消息，昌平实验室与华为在9月17日联合发布TorchX，推出全原子生物分子模型体系，为AI for Science领域提供了一套开源的全原子生物分子设计平台。

据介绍，TorchX是首个基于NPU原生的开源全原子生物分子模型体系，推理与训练代码全面适配昇腾NPU，模型权重按照MIT许可开放，推理代码、训练代码与训练数据也已同步开放。

体系包含三个核心模块：TorchFold用于抗体—抗原复合物结构预测，TorchScore用于候选复合物的结构评价，TorchCraft则通过预测器反演开展分子设计。

TorchFold保留了AlphaFold3的网络架构，并通过界面特异性坐标损失、高噪声扩散调度策略以及两轮置信度筛选蒸馏三项训练策略，把非冗余抗体—抗原训练样本从约5000个扩展到24500个，FoldBench成功率由34.0%提升至70.1%。

TorchScore可跳过扩散生成模块，直接通过置信度通路评价候选复合物；TorchCraft实现了跨分子类型的统一全原子设计，可用于微型蛋白结合体、框架约束VHH、环肽以及小分子结合口袋的设计。

在算力层面，双方通过蛋白质序列残基注意力算子融合、流水调度与昇腾亲和性优化，提升了推理训练速度并降低显存占用。针对长序列场景，方案已在昇腾上完成单卡4K级、整机10K级序列的推理验证，37aa至2002aa测试用例的单卡推理性能提升1.7倍以上，四卡并行可获得3倍以上的性能提升。

TorchX运行于华为智算实验室方案构建的算力底座之上。双方表示，后续将继续推动相关AI工具从实验室走向生命科学产业应用。

注：本文中包含AI辅助创作的内容。

编辑：林一舟