Hugging Face原生支持TorchTPU
发生了什么
Hugging Face宣布原生支持TorchTPU,使Transformers与Diffusers可直接在TPU上运行。用户可通过"tpu"设备选择模型,并使用pip安装torch-tpu。示例代码展示了如何在TPU上加载和运行Qwen/Qwen3.5-9B模型及StableDiffusion3Pipeline。原生支持包括Pallas内核、分布式训练和torch.compile集成,提升模型在TPU上的性能和易用性。
报道与来源
1 篇发布方数量不等于独立核验数量。同稿的多个入口保留在对应报道中。
- Hugging Face Blog一手材料
Hugging Face ❤️ TorchTPU: TPUs, Natively
来源关系与证据边界
按已记录的来源角色展示;连线仅表示原生身份流程确认的同稿入口,不表示独立佐证或跨报道引用。
- 一手材料1 个发布入口