Google 开源 TorchTPU 以支持 AI 应用迁移至 TPU

Google 周二宣布开源其 PyTorch 版本 TorchTPU，允许 AI 应用在无需重写代码的情况下通过简单配置迁移至 Google Cloud 的 TPU 芯片运行。此前一天 Amazon 也发布了类似工具 TorchNeuron 以支持 Trainium 芯片，两家巨头正试图削弱 Nvidia GPU 在 AI 开发领域的默认优势地位。