跳到正文
热点事件持续更新

Hugging Face 发布 TorchTPU,PyTorch 原生支持 TPU

1 篇报道1 个报道来源18 小时前更新

先了解这件事

AI 综述

Hugging Face 发布 TorchTPU,一个原生 PyTorch 后端,使 TPU 成为普通设备。Transformers v5 和 Diffusers 已原生支持 TPU,只需设置 device_map="tpu" 或调用 pipe.to("tpu") 即可运行。 在 TPU VM 上安装只需一条命令:pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu,该索引提供 CPU 版 torch。模型默认以 eager 模式运行,在 TPU 上运行的测试中 96.5% 通过。 Kernels Hub 提供用 Pallas 编写的 TPU 内核,在 Qwen3-30B-A3B 上解码速度提升 21.5 倍。

AI 根据报道生成 · 1 小时前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月20日
  1. Hugging Face Blog精选
    Hugging Face 推出 TorchTPU,为 PyTorch 提供原生 TPU 支持

    Hugging Face 发布 TorchTPU,一个原生 PyTorch 后端,让 TPU 成为普通设备。Transformers v5 和 Diffusers 已原生支持 TPU,只需 device_map="tpu" 或 pipe.to("tpu") 即可运行。Kernels Hub 提供 Pallas 编写的 TPU 内核,在 Qwen3-30B-A3B 上解码速度提升 21.5 倍。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。