产品

PyTorch 宣布原生支持 AWS Trainium 芯片,无需改代码

精选理由

PyTorch 现在能在 Amazon 自研的 Trainium 芯片上直接跑,代码一行不用改,想省算力成本的可以关注。

PyTorch 官方宣布可以原生运行在 AWS Trainium 上,不需要修改代码,也不用学习新框架。Trainium 是 Amazon 自研的 AI 芯片系列,面向深度学习训练和推理负载。在 PyTorch Conference North America 主题演讲中,Amazon Annapurna Labs 的 VP Distinguished Engineer Robert Hundt 将介绍 AWS 为达到性能和质量目标所做的支持。

原文 · PyTorch

PyTorch runs natively on @awscloud Tranium with no code changes. AWS Tranium is @amazon's portfolio of AI chips designed for next-generation deep learning training and inference workloads. You can use it with the tools and workflows you are already familiar with. No new frameworks to learn. No code to change. During PyTorch Conference North America keynote, Robert Hundt, VP Distinguished Engineer, Annapurna Labs, Amazon will share the story and the capabilities AWS provided to meet performance and quality goals. @AmazonScience

#PyTorchCon