Amazon Science·· 14 天前AI 评分55
Amazon 发布 TorchNeuron：面向 AWS Trainium 的开源 PyTorch 后端
The science behind TorchNeuron, Amazon’s new PyTorch backend for AWS Trainium chips
AI 导读
Amazon 本周宣布开源后端 TorchNeuron 可用，它把 PyTorch 代码翻译成可在 AWS Trainium 上执行的操作，已在其他加速器上运行的 PyTorch 代码几乎无需改动即可迁移。
来源：Amazon Science · amazon.science