谷歌联手Meta优化PyTorch适配挑战英伟达

快速阅读: 据相关报道，谷歌母公司Alphabet正推进“TorchTPU”计划，提升其TPU对PyTorch框架的兼容性，以降低开发者迁移成本，并通过与Meta合作及优化第七代TPU v7，加速AI基础设施商业化落地，打破英伟达在AI芯片领域的生态壁垒。

为打破英伟达在AI芯片领域的主导地位，谷歌母公司Alphabet正推进名为“TorchTPU”的战略计划。该计划旨在大幅提升其张量处理单元（TPU）对主流开源框架PyTorch的兼容性，降低开发者从英伟达GPU迁移至TPU的技术门槛与成本。

长期以来，PyTorch与英伟达CUDA软件栈深度绑定，构筑起后者强大的生态壁垒。而谷歌TPU此前主要适配自研的JAX框架，导致大量PyTorch用户在调用TPU时遭遇性能瓶颈。为此，谷歌将投入更多资源，优化底层软件与PyTorch的协同效率。

此外，谷歌正考虑开源部分核心软件组件，以吸引更广泛的开发者群体。据悉，该公司已与PyTorch主要维护方Meta展开深度合作，探讨后者获取更多TPU算力资源的可能性。在硬件方面，谷歌最新发布的第七代TPU v7（代号Ironwood）已针对AI推理场景进行显著优化。

通过TorchTPU补齐软件生态短板，谷歌有望为企业用户提供更具性价比的AI芯片替代方案，并加速其AI基础设施的商业化落地。

(以上内容均由Ai生成)

引用自：AIbase人工智能资讯平台