Google发布TorchTPU,这是PyTorch框架的定制版本,旨在让开发者更便捷地在谷歌张量处理单元(TPU)上构建和运行AI模型。文章指出,PyTorch由Meta于2016年创建,目前超过85%的AI应用基于该框架开发,且长期深度绑定英伟达GPU,这构成英伟达在AI算力市场主导地位的关键原因之一。TorchTPU允许开发者仅通过简单配置调整即可将现有应用迁移至TPU,无需重写底层代码。谷歌并非唯一挑战者,亚马逊AWS此前已推出类似工具TorchNeuron,并向PyTorch上游贡献解耦英伟达GPU的代码。谷歌则通过与Meta、Hugging Face及vLLM、SGLang开发团队合作,确保工具链兼容性。此举反映出AI芯片厂商正从软件生态层面争夺开发者,以削弱英伟达的锁定效应。
谷歌与Meta联手开源TorchTPU项目,为Cloud TPU提供原生PyTorch后端,让PyTorch模型无缝迁移至TPU运行,推理与训练性能直追行业顶尖水平。
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅