AWS Trainium 芯片推出 PyTorch 后端 TorchNeuron
精选理由
如果你在 AWS 上用 PyTorch 训练模型，现在可以直接跑在 Trainium 芯片上，不用换框架，省下不少算力成本。
多数 AI 开发者使用 PyTorch 构建模型。Amazon Science 本周发布 TorchNeuron，为 AWS Trainium 芯片提供 PyTorch 后端。开发者可以继续用 PyTorch 写代码，训练任务运行在 Trainium 硬件上。
原文 · PyTorch
Most AI developers use PyTorch to build their models. And this week @AmazonScience announced a new PyTorch backend for AWS Trainium chips - TorchNeuron. Learn more about the science behind it.
