Open Source2026-10-24Hugging Face Blog

Hugging Face 接入原生 TorchTPU,PyTorch 跑谷歌 TPU 不用再手写胶水代码

Hugging Face 和 TorchTPU 项目正在把 Google TPU 的原生 PyTorch 支持接入 Hugging Face 生态。这次集成想解决的是一个很实际的问题:开发者如果想在 TPU 上训练和运行开源模型,过去得自己写胶水代码,或者做大量移植工作,现在这一步有望省掉。 对那些本来就在用 PyTorch 和 Hugging Face 工具链的团队来说,TPU 会变得更好上手,也等于多了一个跑大规模 AI 负载时替代 GPU 的实用选项。 主流框架的支持之所以重要,是因为基础设施的竞争有一部分就压在“可移植性”上。模型如果能在不同硬件平台之间更容易迁移,开发者手里的牌就更多,可以按成本、可用性和性能来做取舍。TPU 长期以来主要用在谷歌内部和部分客户手里,框架支持面一宽,谷歌体系之外的采用率有可能被拉起来。 对 Hugging Face 来说,这步棋强化了它作为开源模型与工具“中立枢纽”的角色,也表明它想同时支持多种加速器,而不是把整个生态绑在某一家厂商身上。 实际影响还要看几件事:集成跑得顺不顺、能给出多少性能、以及热门库会不会跟进。但原生支持本身已经是个有分量的进展——它降低了尝试 TPU 所需的专业知识门槛,可能带来更多实验。在 AI 训练和推理成本依然是头号焦虑的当下,更容易用上替代硬件,开发者和企业大概都会欢迎。

相关资讯