AI风向标

AWS 发布 TorchNeuron:Trainium 的原生 PyTorch 后端

AWS 宣布 TorchNeuron,一个面向 Trainium 芯片的开源原生 PyTorch 后端,10 月 20 日起以 Apache 2.0 协议公开 Beta,代码位于 github.com/aws-neuron/torch-neuronx。

详细介绍

AWS 宣布 TorchNeuron,一个面向 Trainium 芯片的开源原生 PyTorch 后端,10 月 20 日起以 Apache 2.0 协议公开 Beta,代码位于 github.com/aws-neuron/torch-neuronx。

AI HOT 详情:https://aihot.news/items/szf3v3re4888fzekwoohhu8b4

原文链接:https://pytorch.org/blog/introducing-torchneuron-native-pytorch-backend-for-aws-trainium/