9月17日，昌平实验室携手华为联合发布全原子生物分子模型体系TorchX。该体系为NPU原生的开源平台，推理与训练代码全面适配昇腾NPU，模型权重按MIT许可开放，是首个面向AI4S的NPU原生生物分子设计平台。
TorchX包含三大模块：TorchFold负责抗体–抗原复合物结构预测，TorchScore进行候选复合物结构评价，TorchCraft通过预测器反演开展分子设计。其中TorchFold保留AlphaFold3网络架构，通过三项创新训练策略将非冗余训练样本由约5000个扩展至24500个，FoldBench成功率从34.0%提升至70.1%。
作为业界首个基于昇腾NPU原生的全原子结构预测开源模型，TorchFold通过算子融合、流水调度与昇腾亲和性优化，显著提升推理训练速度并降低显存占用。针对长序列推理中二维Pair激活张量的显存与性能瓶颈，华为团队提出昇腾亲和长序列推理优化方案，已在昇腾上完成单卡4K级、整机10K级序列推理验证，测试用例中单卡性能提升1.7倍以上，四卡并行获得3.x倍提升。
TorchX运行于华为智算实验室方案构建的自主算力底座之上。昌平实验室已开放推理代码、训练代码、模型权重及训练数据，华为将持续聚焦科研核心场景，助力国内上百万科研工作者更高效地开展研究。