返回资讯

谷歌开源 TorchTPU：TPU 原生 PyTorch 后端

发布时间 2026-10-21 12 小时前
来源 Hacker首页帖子
字数 65 字
查看原文

打开项目

这个项目是干什么的

谷歌开源的 PyTorch 后端，让 PyTorch 模型和算子能直接跑在 TPU 上，自带自定义 ATen 内核和编译缓存。

适用人群

适合手上有 TPU、想把 PyTorch 模型搬上去训练或推理的人。

分类

深度学习框架 / TPU 后端 / 分布式训练