更多项目
项目

Google 开源 TorchTPU:TPU 的原生 PyTorch 后端

Google open-sources TorchTPU, a native PyTorch back end for TPUs

AI 导读

该项目为 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
社区讨论
1 帖

项目信息

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

上手方式

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

摘自 README(新标签页打开)