跳到正文
InfoQ · AI/ML· Andrew Hoblitzell·· 3 小时前精选AI 评分69

Google与Meta开源TorchTPU:原生PyTorch支持TPU

PyTorch Conference 2026: Google and Meta Open-Source TorchTPU for Native PyTorch on TPUs

AI 导读

Google与Meta在PyTorch Conference 2026上开源TorchTPU,作为PyTorch在TPU上的原生后端,替代旧的PyTorch/XLA,支持通过model.to("tpu")直接迁移模型。

推荐理由

原文给出了TorchTPU的开源细节和对现有PyTorch生态的兼容性,读者可以据此判断其在多云部署中的实际可用性。

来源:InfoQ · AI/ML · infoq.com