arXiv · AI· Claudio Basile, Michael Voznesensky, Gunhyun Park, Ben Wilson, Parker Holloway, Zhanyong Wan, Vlad Belous, Manish Khanna, Paul Ganssle, Kevin Gleason, Andrew Dame, Brendan Folie, Bhavya Bahl, Haifeng Jin, Pedro Goncalves Mokarzel, Arian Arfaian, Luke Boyer, Chi Zhang, Yiling Yuan, Adam Mainz, Aditya Patel, Huangze Tang, Jay Gan, Lizzie Liang, Will Froom, Tushar Agarwal, Karlo Basioli, Theotime Combes, Emilio Cota, Minos Papadopoulos, Tudor Pipernea, Yarong Mu, Qi Zhou, Yaoshiang Ho, Aleksey Vlasenko, Alexander Brauckmann, Ziyin Huang, Jack Oh, Jerry Hsu, Tanvi Bajpai, Jialei Chen, Wenjun Gu, Chintan Shah, Chris Chan, Sayce Falk, Chris Jones, Rob Torres, Kat Ko, Lee Howes, Freya Azad, Andi Gavrilescu, Joe Pamer, Bill Jia·· 10 小时前AI 评分29
TorchTPU：Google 与 Meta 联合发布的 TPU 原生 PyTorch 开源后端
TorchTPU - A Hybrid Eager/Compiled Architecture for Native PyTorch Execution on TPU
AI 导读
Google 与 Meta 合作推出开源项目 TorchTPU，实现 PyTorch 在 Google TPU 上的原生运行。 该项目通过 eager 执行优化、XLA 集成及硬件加速，提供开箱即用的模型迁移与具有行业竞争力的推理及训练性能。
来源：arXiv · AI · arxiv.org