PyTorch:Blog·· 15 天前精选AI 评分67
AWS 发布 TorchNeuron:Trainium 的原生 PyTorch 后端开源
Introducing TorchNeuron: Native PyTorch backend for AWS Trainium
AI 导读
AWS 宣布推出 TorchNeuron,一个面向 Trainium 的开源原生 PyTorch 后端,10 月 20 日以 Apache 2.0 协议在 PyTorch Conference North America 开启公测。
推荐理由
原文给出 TorchNeuron 的架构、执行模式和兼容性测试数据,开发者可据此评估把现有 PyTorch 工作流迁到 Trainium 的成本。
来源:PyTorch:Blog · pytorch.org