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