AWS 说，PyTorch 现在可以在 Trainium 上原生运行： 不用学新框架。 不用重写训练代码。 继续使用熟悉的 PyTorch 工作流。 听起来… · @RwatitFahsa · TwKit

@RwatitFahsa

AWS 说，PyTorch 现在可以在 Trainium 上原生运行： 不用学新框架。 不用重写训练代码。 继续使用熟悉的 PyTorch 工作流。 听起来像是把 .to("cuda") 换成 .to("neuron") 就能结束战斗。 但目前还不能把“零改动”理解得太绝对。 不少项目仍会直接调用 torch.cuda.*，部分算子、编译速度和生态成熟度也还追不上 CUDA。 不过方向已经很明确： AWS 不只是想做一块更便宜的 AI 芯片，而是想让开发者换掉英伟达时，感觉不到自己换了平台。 芯片能不能赢，最后可能要看迁移时要改多少行代码。

曝光 24 · 评论 0 · 点赞 0 · 书签 0