AniSignal
← AI 热点

1 个来源 · 1 篇报道 · 2026/10/20 08:00

Hugging Face 发布 TorchTPU 支持原生 TPU 后端

Hugging Face 发布 TorchTPU，作为 PyTorch 的原生 TPU 后端，允许用户在 TPU 上直接运行 Transformers、Diffusers 和 Kernels 模型。用户可通过 device_map="tpu" 或 pipe.to("tpu") 将模型发送至 TPU，该操作在 TPU VM 上可通过一条 pip 命令安装。

最新进展：Hugging Face 发布 TorchTPU：PyTorch 原生支持 TPU 设备

事实与进展

  1. Hugging Face 发布 TorchTPU：PyTorch 原生支持 TPU 设备

    Hugging Face 发布 TorchTPU，作为 PyTorch 原生 TPU 后端，允许用户通过 device_map="tpu" 或 pipe.to("tpu") 直接在 TPU 上运行 Transformers 和 Diffusers 模型。

    Hugging Face Blog · 阅读报道 →

报道时间线