当护城河变成入场券:TorchTPU 与 CUDA 锁定的黄昏
从一个 .to("tpu") 说起 上周的 PyTorch Conference North America 2026 上,Google 与 Meta 联合开源了 TorchTPU——一个面向 Google TPU 的原生 PyTorch 后端。官方的口号异常克制:把 model.to("cuda") 改成 model.to("tpu"),你的模型就能在 TPU 上跑起来。 对一个不熟悉基础设施的人来说,这条新闻听起来像是无数个"XX 硬件支持 PyTorch"公告中的一个。但把它放进过去几年 AI 算力格局的脉络里看,它的分量远超表面:这是 CUDA 生态锁定被撬动的过程中,迄今最实质性的一块砝码落地的声音。 这篇文章想回答三个问题:TorchTPU 到底解决了一个什么问题?为什么出手的是 Google 和 Meta 这对奇怪的组合?以及,“CUDA 锁定松动"对普通的 AI 工程团队意味着什么? 一、被 XLA 挡在门外的 PyTorch 用户 要理解 TorchTPU 的价值,得先理解它取代了什么。 TPU 是为机器学习定制的芯片,软件层之下由 XLA 编译器栈驱动。历史上,TPU 与 TensorFlow、JAX 这两个 Google 亲儿子框架深度绑定。PyTorch 用户想用 TPU,走的是 2020 年 Google 推出的 PyTorch/XLA——一条基于 Lazy Tensor 的间接编译路径。 问题在于,PyTorch/XLA 本质上是把 PyTorch “硬塞进” JAX 的工作模式: ...