# Google 开源 TorchTPU：面向 TPU 的原生 PyTorch 后端

> 原标题：Google open-sources TorchTPU, a native PyTorch back end for TPUs

- 来源：Hacker News 首页
- 发布时间：2026-10-20T16:40:24.000Z
- AX AI 日报：https://ai-daily.ax0x.ai/items/aw71ahcgs36yvhz5pah8vsqhs
- 原文：https://github.com/google-pytorch/torch_tpu

## 摘要

Google 开源 TorchTPU，为 TPU 提供原生 PyTorch 后端，让 PyTorch 模型和算子可直接在 TPU 设备上运行。项目包含自定义 ATen kernel、XLA 计算编译缓存，支持数据并行与张量并行分布式训练，并集成 torch.compile() 图模式执行。