❯ Google 开源 TorchTPU，让 PyTorch 工作负载更容易迁移到 TPU

训练框架增加硬件选项Google 10 月 20 日宣布开源 TorchTPU，这是与 Meta 的 PyTorch 团队共同开发的原生 TPU 支持方案。PyTorch 是广泛使用的模型开发框架，TPU 是 Google 面向机器学习设计的计算芯片；新工具让开发者以较少代码修改，将现有 PyTorch 工作负载迁移过去。

保留熟悉的调试方式TorchTPU 支持逐步执行和调试，再通过 torch.compile 将程序编译优化，用于正式运行。Google 还将它接入 Hugging Face，以及 vLLM、SGLang 等模型服务工具，使开发、训练和部署可以沿用较多现有流程，减少更换硬件时重新搭建软件环境的工作。

竞争从芯片延伸到工具链TPU 的吸引力取决于芯片表现，也取决于团队能否方便地迁移、排错和维护。原生 PyTorch 支持把硬件选择纳入更熟悉的开发环境，企业可以更直接地评估不同平台的服务成本。Google 的这次开放，将软件兼容性放到了算力竞争的中心。

▮ SIGNAL当迁移和维护成本下降，企业才更容易把芯片报价与实际服务成本放在一起比较，硬件替代也才有更多选择。