Google 联手 Meta 开源 TorchTPU,PyTorch 原生支持 TPU

在 PyTorchConf 2026 上,Google 与 Meta 官方开源 TorchTPU,作为 PyTorch 在 Google TPU 上的原生后端,由 Google 于 2025 年启动开发。现有 PyTorch 代码无需修改即可在 TPU 上运行,TorchTPU 还被集成进 TorchTitan,使 Llama 3 等模型可直接在 Google Cloud TPU 上进行大规模预训练,显著降低了使用门槛。

2026-10-22 ~ 2026-10-23 · 2 条相关