产品发布 / 更新精选
TorchTPU:在谷歌规模上原生运行 PyTorch 于 TPU
原始标题:TorchTPU: Running PyTorch Natively on TPUs at Google Scale
内容摘要
TorchTPU 是一个新的工程栈,旨在让 PyTorch 工作负载以最小代码改动在谷歌 TPU 基础设施上获得原生高性能体验。它采用"Eager First"设计,提供多执行模式,并利用 XLA 编译器优化大规模集群的分布式训练。项目计划到 2026 年进一步降低编译开销,扩展对动态形状和自定义内核的支持,以确保下一代 AI 实现无缝扩展。
内容分类AI 产品发布与更新
内容层级精选情报
发布时间(北京时间)
本站收录时间(北京时间)
信息来源Google Developers Blog(RSS)
站内情报编号intel-85a908cf148901465363eee2