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

- 原标题：Google open-sources TorchTPU, a native PyTorch back end for TPUs
- 类型：项目
- 来源：GitHub
- 发布时间：2026-03-10
- 首次精选：2026-10-26 15:43
- 分级：值得看
- 原文链接：[https://github.com/google-pytorch/torch\_tpu](<https://github.com/google-pytorch/torch_tpu>)
- 本站页面：[https://oplearnx.com/resources/pk\_a66b7d74523f](<https://oplearnx.com/resources/pk_a66b7d74523f>)

## 摘要

该项目为 Google TPU 提供 PyTorch 后端，使 PyTorch 模型和算子可在 TPU 上运行，包含面向多种 PyTorch 操作的 ATen 内核、用于优化 XLA 计算的编译缓存，以及分布式训练（数据并行、张量并行）和模型推理示例。列出支持的示例模型包括 Llama、Qwen、ResNet、DLRM、minGPT、UNet，并提供 torch.compile\(\) 集成、调试与基准工具。文档为 Marimo 交互式笔记本，需在 TPU VM（如 v6e GCE 实例）上运行。

## 热度

- 档位：热度强
- 7 天新增 star：0
- 30 天新增 star：0
- Hacker News：[379 分 · 92 条评论](<https://news.ycombinator.com/item?id=50169053>)
- 社区讨论：1 帖

## 项目信息

- 语言：C\+\+
- 许可：未声明
- 最近提交：2026-11-01
- Star：804
- Fork：77

## 上手方式

```bash
python -m venv .venv
source .venv/bin/activate
```

摘自 [README](<https://github.com/google-pytorch/torch_tpu/blob/main/README.md>)

---
来源：LearnMap · 学径（https://oplearnx.com）