跳到正文
智迹闻AuraTracer

Hugging Face原生支持TorchTPU

本机阅读

发生了什么

Hugging Face宣布原生支持TorchTPU,使Transformers与Diffusers可直接在TPU上运行。用户可通过"tpu"设备选择模型,并使用pip安装torch-tpu。示例代码展示了如何在TPU上加载和运行Qwen/Qwen3.5-9B模型及StableDiffusion3Pipeline。原生支持包括Pallas内核、分布式训练和torch.compile集成,提升模型在TPU上的性能和易用性。

报道与来源

1 篇

发布方数量不等于独立核验数量。同稿的多个入口保留在对应报道中。

  1. Hugging Face Blog一手材料

    Hugging Face ❤️ TorchTPU: TPUs, Natively

来源关系与证据边界

按已记录的来源角色展示;连线仅表示原生身份流程确认的同稿入口,不表示独立佐证或跨报道引用。

  1. 一手材料1 个发布入口