亚马逊发布 TorchNeuron:PyTorch 代码零改动跑上 Trainium 芯片

TheNickWalsh · x · 2026-10-19

亚马逊推出开源工具 TorchNeuron,开发者无需重写代码即可将现有 PyTorch 代码直接运行在 AWS Trainium 芯片上。由于约 85% 的 AI 研究者和开发者使用 PyTorch,该工具让现有的测试、调试和扩缩容工具链在 Trainium 上原样可用,显著降低采用新硬件的门槛。从语音 AI 到药物发现模型的多支团队已在实际使用。这被视为 AWS 降低 Trainium 采用摩擦、与 NVIDIA CUDA 生态竞争的重要一步。

所属事件:AWS 开源 TorchNeuron,PyTorch 零改动跑上 Trainium(3 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →