9月17日晚间消息,昌平实验室携手华为联合发布TorchX,正式推出全原子生物分子模型体系。TorchX是业界首个NPU原生的开源全原子生物分子模型体系,推理与训练代码全面适配昇腾NPU,模型权重按MIT许可开放。此次联合发布为全球AI4S提供了首个NPU原生的开源生物分子设计平台,打造“自主创新+前沿模型”的技术底座,对降低生物医药领域AI应用门槛、促进科研资源共享和成果可复现具有重要示范意义。
TorchX包含三个核心功能模块:TorchFold用于抗体–抗原复合物结构预测,TorchScore用于候选复合物的结构评价,TorchCraft通过预测器反演开展分子设计。
其中,TorchFold保留AlphaFold3的网络架构,通过界面特异性坐标损失、高噪声扩散调度策略及两轮置信度筛选蒸馏三项创新训练策略,将非冗余抗体–抗原训练样本由约5000个扩展至24,500个,FoldBench成功率由34.0%提升至70.1%。
TorchCraft中使用了可重入式梯度重计算,期间昌平实验室与华为团队联合修复了Torch原生梯度重计算在当前模型下的显存泄漏问题。针对TorchCraft长序列场景,华为团队对FA算子进行PSE反向计算功能补齐,使得当前模型下该算子可以正常使能,在显存优化和性能提升上均有较好的表现。
TorchScore作为结构评分模式,可跳过扩散生成模块直接通过置信度通路评价候选复合物。TorchCraft则实现了跨分子类型的统一全原子设计,可同时用于微型蛋白结合体、框架约束VHH、环肽和小分子结合口袋设计。针对IFNAR2、IL-7Rα等靶点的微型蛋白及VHH设计均经生物层干涉法验证达到纳摩尔亲和力,且实验候选直接取自TorchCraft输出。
全原子生物分子模型体系TorchX运行于华为智算实验室方案所构建的自主算力底座之上。TorchX的发布不仅是技术突破,更是生态共建的开端。昌平实验室已开放推理代码、训练代码、模型权重以及训练数据。华为也将聚焦科研核心场景,助力国内上百万科研工作者通过AI更加高效地进行科研。未来,双方将共同推动AI工具从实验室走向生命科学产业应用,为新药发现、新蛋白设计及未来满足医学需求相关研究提供可复现、可扩展的基础模型支撑。
