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

Hugging Face 接入 TorchTPU 原生 TPU 后端

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

先了解这件事

AI 综述

Hugging Face 宣布将 TorchTPU 作为 PyTorch 原生后端接入其生态,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行。加载模型时传入 device_map="tpu" 或调用 pipe.to("tpu") 即可,不再需要 Optimum TPU。 据 Hugging Face 博客,在 TPU VM 上安装只需一条命令 pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu,该索引提供的是 CPU 版 torch;模型默认以 eager 模式运行。博客称,在 TPU 上运行的测试中 96.5% 通过。

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

报道时间线

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

10月20日
  1. Hugging Face Blog精选
    Hugging Face 接入 TorchTPU,Transformers 与 Diffusers 原生支持 TPU

    Hugging Face 宣布 TorchTPU 作为 PyTorch 原生后端接入其生态,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,加载模型时传入 device_map="tpu" 或 pipe.to("tpu") 即可,不再需要 Optimum TPU。

本事件热度走势

可比范围当前
6
可比范围峰值
610月21日 00:00
近 24 小时变化
–

趋势仅比较持续完整观测到的相同主体,范围可能小于当前热度统计。移动指针或点击图表查看每小时热度;键盘可用左右方向键切换。