Event ·Multi-source

Google 开源 TorchTPU,让 PyTorch 代码脱离 Nvidia GPU 跑在 TPU 上

Google 开源了 TorchTPU,用它替代 PyTorch/XLA 这一层。此前 TPU 依赖 PyTorch/XLA 把操作排队交给编译器,调试方式与 GPU 不同。TorchTPU 让每个操作按原样逐步运行,方便调试,生产环境再切到 torch.compile 提速。实测训练 Qwen 3 8B 时最高用到 TPU 峰值算力的 52%,Amazon 也提供类似的 PyTorch 原生后端。

Current read

Google 把 TorchTPU 开源了,PyTorch 代码不用改就能跑 TPU,调试还能单步执行,想摆脱 Nvidia 锁定的可以试试。

11 sources67° AI heatLast updated 2026/10/20 19:31:43

Evidence

11 sources
Primary sourcerohanpaul_ai
View source
Related source 1techcrunch
View source
Related source 3VTV Công nghệ
View source
Related source 4marktechpost
View source
Related source 5TestingCatalog
View source
Related source 6kimmonismus
View source
Related source 7Teortaxes
View source
Related source 9AI21 Labs
View source
Related source 10@atomic_chat_hq
View source

Conflict check

Dedup data only shows that these sources discuss the same event, not that they agree. There is no structured supporting or refuting evidence yet, so no conflict is inferred automatically.