文章摘要
国内AI芯片企业寒武纪近期成为PyTorch基金会最高级别白金会员,获得理事会席位,这是中国芯片厂商首次跻身全球核心AI开发框架决策层。此前中国AI芯片在全球开源生态中多为适配者,寒武纪长期坚持向上游社区贡献核心代码,此次加入标志中国AI芯片完成从适配者到全球AI开源生态核心参与者的转型。

就在近期,国内AI芯片企业寒武纪正式成为PyTorch基金会最高级别白金成员,拿到了基金会理事会的席位。这意味着,在全球AI开发框架的决策房间里,首次出现了中国芯片厂商的身影。

很多开发者熟悉PyTorch,但未必了解其背后的基金会。作为Linux基金会旗下的社区驱动开源AI组织,PyTorch基金会掌控着PyTorch、vLLM、DeepSpeed、Ray等一众核心AI基础设施项目。当前全球主流的大模型开发与部署,几乎都离不开这套工具链,从各类开源大模型到商用生成式AI产品,底层都依托PyTorch完成模型构建,借助vLLM等推理引擎实现高效部署。截至2026年,PyTorch社区已经汇聚了来自2000多家机构的1.2万名贡献者,这套工具早已不是单一公司的产品,而是整个AI行业的通用地基。

基金会的理事会,正是这块地基的设计者。理事会成员不需要讨论抽象的技术愿景,而是要决定PyTorch下一步优先支持哪些硬件、设备接口如何抽象、哪些能力会进入主干代码、哪些功能留给厂商自行实现。2022年基金会成立时,创始理事会成员清一色都是全球科技巨头,包括Meta、AMD、AWS、Google Cloud、Microsoft、NVIDIA。此次寒武纪拿到的白金席位,属于该层级的最高档位,意味着寒武纪拥有了和这些巨头同等的决策话语权。

长期以来,中国AI芯片在全球开源生态中的角色始终是“适配者”:其他厂商先完成框架开发,国内企业需要做兼容、补全算子、调优性能,尽可能让开发者已有的代码能够在自家芯片上运行。这种工作既辛苦又难以获得广泛认可,框架升级一次就要跟进一次,硬件迭代一次就要重构一次软件栈,最终得到的评价往往只是“能用”,却很难做到“好用”。更关键的是,框架的设计逻辑、接口定义这些核心规则,始终由海外厂商主导,国内芯片厂商没有参与制定的机会,开发者更换硬件往往需要重写大量代码,这也成为了国产芯片推广的一大障碍。

寒武纪选择了一条截然不同的路径。他们坚持Upstream First的原则,也就是将所有代码改动优先提交到PyTorch的公共主干仓库,而非保留私有分支。这条道路远比直接做私有分支要艰难:需要经过社区的严格评审,要和不同硬件厂商的诉求进行妥协,一个代码补丁往往需要数月才能完成合并。但这种方式能够彻底摆脱“永远慢半拍”的困境,上游框架的每一次更新都能直接同步到产品中。

经过多年的积累,寒武纪已经向PyTorch主干提交了覆盖七个核心模块的代码,包括torch编译器、Eager模式算子、设备运行时、分布式计算、自动混合精度、数据加载器以及Profiler性能分析器。简单来说,他们参与了模型编译、数据流转、显存优化、多卡通信、性能调试等全流程的底层设计。同时,寒武纪还和vLLM社区展开深度合作,实现了主流开源大模型的Day 0 适配,也就是模型正式发布当天,就能在寒武纪的芯片上流畅运行。对于需要赶产品周期的团队来说,这三个月的差距往往决定了产品能否抢占市场窗口。

此次寒武纪加入基金会,派出的并非商务或高管,而是两位核心技术人员。王进,寒武纪AI框架高级总监,代表公司进入理事会;朱靖,寒武纪PyTorch项目的核心维护者,加入了基金会的技术咨询委员会。一个负责顶层决策,一个聚焦一线技术对接,这两个席位清晰传递出寒武纪的目标:不只是“被PyTorch支持”,而是要参与决定PyTorch如何支持所有硬件开发者。

寒武纪软件工程副总裁Elton Gong表示,统一的框架接口能够降低开发者的迁移成本,提供一致的工具和开发体验;对于PyTorch来说,更通用的设备抽象则能减少下游分支,让项目维护更可持续。寒武纪的目标不止是让自家产品支持PyTorch,而是通过持续的主干共建,推动PyTorch更好地适配多设备后端,让全球开发者都能在更广泛的硬件平台上获得开箱即用的原生开发体验。

开发者更换硬件时,最头疼的就是重写代码、重新学习工具链、反复调优性能,这种顾虑让很多团队不敢轻易更换硬件。寒武纪向PyTorch主干提交代码的行为,本质上就是在推动硬件生态的统一,让更换芯片就像更换外设一样简单,不需要重新学习一套全新的开发流程。PyTorch基金会执行董事Mark Collier也对此表示认可,他提到寒武纪持续为上游社区贡献力量,同时推动vLLM在寒武纪硬件上的应用,欢迎其加入基金会。

寒武纪加入PyTorch基金会,绝非一次简单的会员身份变更,而是这家成立于2016年的AI芯片企业,从“国产AI芯片厂商”向“全球主流AI软件生态参与者”转型的标志性节点。十年间,寒武纪的芯片已经在智能对话、智能体AI、搜索广告推荐、音视频生成等场景实现规模化商用,而软件生态的布局——从坚持Upstream First,到贡献七大主干模块,再到拿下理事会和技术咨询委员会席位——让他们终于拿到了全球AI开源生态的入场券。

从“我们也支持PyTorch”到“PyTorch的多设备能力有我们的一份”,中国AI芯片终于在全球开源生态的牌桌上占据了一席之地。

以上内容不代表本平台立场,仅供读者参考