★ 4
谷歌与 Meta 开源 TorchTPU,让 PyTorch 原生跑在 TPU 上
TorchTPU 是谷歌与 Meta 合作的开源项目,为谷歌 TPU 提供原生 PyTorch 后端,支持开箱即用的模型执行与 TPU 迁移。它通过 eager 执行优化、torch.compile 集成与原生硬件加速,提供具备行业竞争力的推理与训练性能。
arXiv cs.AI13 天前
共 1 篇相关资讯,按重要度与时间排序,每 3 小时更新。
TorchTPU 是谷歌与 Meta 合作的开源项目,为谷歌 TPU 提供原生 PyTorch 后端,支持开箱即用的模型执行与 TPU 迁移。它通过 eager 执行优化、torch.compile 集成与原生硬件加速,提供具备行业竞争力的推理与训练性能。