谷歌开源TorchTPU以挑战英伟达在AI开发中的主导地位
aiinsider ·
Google 已正式开源 TorchTPU,这是其 PyTorch 实现的一项重要更新,旨在促进张量处理单元(TPU)的使用。此举旨在降低行业对 Nvidia GPU 的严重依赖,后者长期以来一直是 AI 开发和研究的默认标准。通过允许应用以最小的配置更改在 Google Cloud 上运行,TorchTPU 使开发人员能够在无需重写代码的情况下切换硬件。此次发布与亚马逊近期的类似举措相呼应,后者最近推出了 TorchNeuron 以支持 AWS 上的 Trainium 芯片。为确保广泛的兼容性,Google 与 Meta、Hugging Face、Inferact 和 RadixArk 等关键行业参与者进行了合作。此外,Google 还推出了 TPU 代理套件,以帮助进行应用迁移和故障排除,并更新了 LiteRT 工具,以更好地支持移动和嵌入式设备。这些举措共同挑战了 Nvidia 在 AI 基础设施市场的主导地位。
aiinsider