Google 与 AWS 推 PyTorch 原生后端进军自研芯片

PyTorch Conference 2026 上,Google TorchTPU 与 Amazon TorchNeuron 两大新后端成为焦点,可让现有 PyTorch 代码几乎零改动地迁移到 TPU 和 Trainium 芯片。AWS 已宣布 TorchNeuron 开源。SemiAnalysis 深度分析 Trainium3 指出,该芯片此前被视为推理专用,如今在训练侧也具备竞争力,原生 PyTorch 支持使其有望叫板 GPU 训练主导地位。调查显示约 85% 的 AI 研究者使用 PyTorch。

2026-10-23 ~ 2026-10-24 · 3 条相关