谷歌开源TorchTPU加速PyTorch应用迁移至TPU
谷歌正式将TorchTPU作为开源项目发布，支持开发者通过调整配置而非重写代码，即可将在PyTorch上构建的AI应用无缝迁移至TPU芯片及Google Cloud平台。此举恰逢亚马逊推出Trainium芯片配套工具TorchNeuron次日，显示出云厂商在AI硬件生态适配领域的激烈竞争。
鉴于主流AI应用多基于Meta于2016年创建的PyTorch框架构建，且该框架早期设计主要围绕Nvidia GPU优化，谷歌已联合Meta、Hugging Face以及vLLM、SGLang等工具开发商开展深度合作，确保TorchTPU与上述平台的兼容性。同时，谷歌还发布了AI代理工具，旨在辅助开发者高效完成向TPU的迁移工作并解决潜在技术问题。