Amazon Science·· 24 天前AI 评分22
Amazon 开源 TorchNeuron：将 PyTorch 代码跑在 AWS Trainium 上的新后端
The science behind TorchNeuron, Amazon’s new PyTorch backend for AWS Trainium chips
AI 导读
Amazon 本周开源 TorchNeuron，这一 PyTorch 后端可将 PyTorch 代码转换为 AWS Trainium 可执行的操作，已有代码几乎无需改动即可运行。
来源：Amazon Science · amazon.science