每天听简报，了解最新科技、AI、软件资讯
PyTorch 现在能不改一行代码，直接跑在亚马逊云科技自研的人工智能芯片 Trainium 上。以前想在这颗芯片上训练模型，得靠专门的编译工具链，把代码转换一遍才能跑起来，改动不小。这次亚马逊把 PyTorch 里逐行执行和先编译再运行这两种运行方式都接了进去，还打通了几个配套的训练框架和抱抱脸的模型库，相关代码也提交回了 PyTorch 的主干仓库。这意味着工程师原来写好的训练脚本，换到这颗芯片上基本不用大改，省掉了一整层适配工作。这些细节会在十月的 PyTorch 开发者大会上，由亚马逊的工程师详细讲述，目前对外公开的只是一场演讲预告。
进度保存在本设备 · 登录同步