产品与应用媒体报道 · pytorch.org
AWS 发布 TorchNeuron:Trainium 的原生 PyTorch 后端
OmniTools 10月19日消息,AWS 宣布推出 TorchNeuron,这是一个面向 AWS Trainium 芯片的开源原生 PyTorch 后端。
该后端将于 10 月 20 日起以 Apache 2.0 协议进入公开 Beta,代码托管在 github.com/aws-neuron/torch-neuronx。
目前公开摘要未披露更多功能细节和性能数据。