PyTorch 博客· Nafea Bshara, Robert Hundt, Ron Diamant, Colin Brace, Rahul Solanki, Maen Suleiman, Khaled ElGalaind, Temesghen Kahsai, Cyril Bortolato, Indu Thangakrishnan, Shauheen Zahirazami, Sarang Ozarde, Adarsh Singh, Hanbo Wang, Esha Lakhotia, Jianying Lang·· 4 小时前AI 评分46
引入 TorchNeuron:AWS Trainium 的原生 PyTorch 后端
Introducing TorchNeuron: Native PyTorch backend for AWS Trainium
AI 导读
PyTorch 博客宣布推出 TorchNeuron,一个为 AWS Trainium 设计的原生 PyTorch 后端,支持现有模型代码、库和工具直接运行。
来源:PyTorch 博客 · pytorch.org