国际大厂·· 2 小时前精选AI 评估 · 80/100分数表示内容值得关注的程度,依据实质份量、信息增量、证据强度、读者相关性和可用性,按内容类型加权。显示分数为两次独立评分的平均值。AI 判断可能有误,评分不代表事实已被核实,也不同于热点榜的讨论热度。
Google 开源 TorchTPU,可在 TPU 上原生运行 PyTorch
TorchTPU is now open source: Run native PyTorch on TPUs
AI 导读
Google 在 PyTorch Conference 2026 上宣布 TorchTPU 在 GitHub 开源,TPU 由此与 GPU、CPU 一样成为 PyTorch 的原生运行选项,现有 PyTorch 工作负载只需少量代码改动即可在 TPU 上训练、服务和部署。
推荐理由
原文列出 TorchTPU 与 vLLM、SGLang、TorchTitan 等框架的对接方式,可据此判断 TPU 在现有 PyTorch 工作流中的接入成本。
来源:国际大厂 · cloud.google.com