跳到正文
热点事件观察中

Hugging Face 接入 TorchTPU,TPU 原生运行

1 篇报道1 个报道来源1 天前更新

先了解这件事

AI 综述

Hugging Face 宣布接入 TorchTPU 这一原生 PyTorch 后端,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,不再需要 torch_xla 或 Optimum TPU。 在 TPU 虚拟机上安装只需一条命令:pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu。Hugging Face 称,附加索引提供 torch 的 CPU 版本;不加则 pip 会拉取体积更大的 CUDA 版本,而 TPU 上并不需要它。 Hugging Face 称,在 TPU 上运行的测试中 96.5% 通过,并正在添加 TPU 上的 nightly CI,结果将公布在 Transformers 仪表盘。

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

报道时间线

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

10月20日
  1. Hugging Face Blog精选
    Hugging Face 支持 TorchTPU:Transformers 与 Diffusers 原生运行在 TPU 上

    Hugging Face 宣布接入 TorchTPU 这一原生 PyTorch 后端,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,不再需要 torch_xla 或 Optimum TPU。

本事件热度走势

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