科技日报记者 崔爽
9月17日，昌平实验室携手华为联合发布TorchX，推出全原子生物分子模型体系。据介绍，TorchX是业界首个NPU（神经网络处理器）原生的开源全原子生物分子模型体系，推理与训练代码全面适配昇腾NPU，模型权重按MIT许可（起源于麻省理工学院的宽松软件许可证）开放。此次联合发布为全球AI4S（人工智能驱动的科学研究）提供了首个NPU原生的开源生物分子设计平台，打造“自主创新+前沿模型”的技术底座，持续降低生物医药领域AI应用门槛，促进科研资源共享和成果可复现。
据了解，TorchX包含三个核心功能模块：TorchFold用于抗体—抗原复合物结构预测，TorchScore用于候选复合物的结构评价，TorchCraft通过预测器反演开展分子设计。
同时，华为与昌平实验室协同创新，通过蛋白质序列残基注意力算子融合、流水调度以及昇腾亲和性优化等方式，显著提升推理训练的速度和降低显存占用。全原子生物分子模型体系TorchX运行于华为智算实验室方案所构建的自主算力底座之上，为TorchX创新筑牢智能底座，进一步提升科研效率，推动科研智能化转型，助力AI4S落地。
TorchX的发布也是生态共建的开端。昌平实验室已开放推理代码、训练代码、模型权重以及训练数据，华为也将聚焦科研核心场景，助力国内上百万科研工作者通过AI更加高效地进行科研。未来，双方将共同推动AI工具从实验室走向生命科学产业应用，为新药发现、新蛋白设计及未来满足医学需求相关研究提供可复现、可扩展的基础模型支撑。
网友评论
