Hugging Face Blog·· 1 天前精选
Hugging Face 发布原生 PyTorch TPU 后端 TorchTPU
Hugging Face ❤️ TorchTPU: TPUs, Natively
研究任务与主要进展
Hugging Face 发布 TorchTPU，为 PyTorch 增加原生 TPU 后端，使 Transformers、Diffusers 和 Kernels 可直接将设备设为“tpu”，无需再依赖 Optimum TPU。
阶段、条件与复现 · 深读核对
- 新能力对应什么具体任务与最小输入输出？
- 代码、模型、数据、许可与可用入口是否明确？
- 效果、总成本、失效条件与实际工作流如何验证？
这些是阅读核对问题；材料未说明的条件保留未知。请阅读原文核验。
本站判断
Hugging Face 宣布 TorchTPU 作为原生 PyTorch 的 TPU 后端，覆盖 Transformers、Diffusers 与 Kernels，并提供 Pallas 内核、分布式训练、torch.compile 和模型分片支持，降低原有 torch_xla 或 JAX 路径的迁移成本。材料中的性能和测试数据属于官方发布说明，当前 TPU 测试通过率为 96.5%，已有 34 个问题报告、19 个修复；其实际价值仍取决于具体模型、TPU 配置与工作负载。
来源：Hugging Face Blog · huggingface.co