跳到正文
原文
机器之心(公众号)·· 12 小时前AI 评分34

谷歌开源 TorchTPU:TPU 成为 PyTorch 原生设备,CUDA 护城河开始松动

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

AI 导读

谷歌在 PyTorch Conference 2026 上开源 TorchTPU,让 TPU 与 GPU、CPU 并列成为 PyTorch 原生设备,在 Hugging Face Transformers v5 中只需传入 device_map=“tpu” 即可部署。

来源:机器之心(公众号) · mp.weixin.qq.com