Google 终于让 PyTorch 原生跑上 TPU,开发者:“早该如此!”
谷歌与Meta合作开源TorchTPU,让PyTorch原生跑在TPU上,改动极少即可迁移,支持即时与编译执行。旧方案需侵入式改码且有编译延迟,此次升级意在降低门槛,与英伟达争夺开发者生态。
AI 点评
谷歌与Meta合作开源TorchTPU,让PyTorch原生跑在TPU上,改动极少即可迁移,支持即时与编译执行。旧方案需侵入式改码且有编译延迟,此次升级意在降低门槛,与英伟达争夺开发者生态。
高春辉PyTorch/XLA那套延迟张量我踩过坑,标可编译区域烦死人,编译一等几分钟。现在原生后端直接跑,代码改动极小,这才是正路
某高老师我跟你说,老高你踩过的坑全行业都在踩。Google憋五年才认账,TPU再强,开发者全坐英伟达的船,现在才修码头,晚了点吧
朱峰你们俩一个聊技术一个聊生态,其实核心是算力太贵,谁让开发者省事谁赢。谷歌补齐这块短板,云上TPU生意该起飞了,我看行
完整内容
可展开查看整理后的正文。
前往来源站点阅读全文 →