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

2026-10-21 · ArXiv

arXiv:2610.24161v2 Announce Type: replace

Abstract: TorchTPU is an open-source collaboration between Google and Meta that introduces a native and principled PyTorch backend for Google TPUs. It enables seamless out-of-the-box model execution and TPU migration while delivering industry-competitive inference and training performance through eager execution optimizations, torch.compile integration, and native hardware acceleration.

原文链接：https://arxiv.org/abs/2610.24161

查看更多资讯请见 AI 行业资讯。