项目
Google 开源 TorchTPU:TPU 的原生 PyTorch 后端
Google open-sources TorchTPU, a native PyTorch back end for TPUs
该项目为 Google TPU 提供 PyTorch 后端,使 PyTorch 模型和算子可在 TPU 上运行,包含面向多种 PyTorch 操作的 ATen 内核、用于优化 XLA 计算的编译缓存,以及分布式训练(数据并行、张量并行)和模型推理示例。列出支持的示例模型包括 Llama、Qwen、ResNet、DLRM、minGPT、UNet,并提供 torch.compile() 集成、调试与基准工具。文档为 Marimo 交互式笔记本,需在 TPU VM(如 v6e GCE 实例)上运行。
- 分级
- 值得看
- 所属主题
- 整理中
- 收录时间
- 2026-10-26
热度
热度强
- 7 天新增 star
- 0
- 30 天新增 star
- 0
- Hacker News
- 379 分 · 92 条评论(新标签页打开)
- 社区讨论
- 1 帖
项目信息
- 语言
- C++
- 许可
- 未声明
- 最近提交
- 2026-11-01
- Star
- 804
- Fork
- 77
上手方式
bash
python -m venv .venv
source .venv/bin/activate