跳到正文
实时
机器之心·收录于· 21 天前78

谷歌开源 TorchTPU,让 PyTorch 原生运行在 TPU 上

谷歌开源TorchTPU!TPU成PyTorch原生设备,CUDA最深的护城河开始松动

要点速览

谷歌在圣何塞举行的 PyTorch Conference 2026 上宣布开源 TorchTPU,这是一套让 PyTorch 原生运行在 TPU 上的完整软件栈,可用 device_map="tpu" 一行代码指定设备。材料提到这一变化涉及 Hugging Face 相关流程,但正文未完整给出。

来源: 机器之心 · jiqizhixin.com