学术研究

TorchTPU-一种在TPU上执行原生PyTorch的混合EAGER/COMPILED体系结构

TorchTPU - A Hybrid Eager/Compiled Architecture for Native PyTorch Execution on TPU

arXiv cs.AI发布于 10月30日 1 分钟阅读

专题配图图片仅供信息参考图片仅供参考版权归原作者

摘要

Arxiv：2610.24161v4公告类型：替换摘要：TorchTPU是Google和Meta之间的开源合作，为Google TPU引入了一个原生的、有原则的PyTorch后端。它支持无缝的开箱即用模型执行和TPU迁移，同时通过急切的执行优化、Torch.Compile集成和本机硬件加速提供具有行业竞争力的推理和培训性能。

原始资料

arXiv cs.AI · 中文编译

查看来源原文产品参数、测试条件及最新修订以来源原文为准。