如果只能用一行代码概括这条新闻,那就是device_map="tpu"。 在圣何塞举行的 PyTorch Conference 2026 上,谷歌宣布开源TorchTPU,这是一套让 PyTorch 原生运行在 TPU 上的完整软件栈。从此,在 ...