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

Hugging Face 接入 TorchTPU,TPU 原生运行

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

先了解这件事

AI 综述

Hugging Face 宣布接入原生 PyTorch 后端 TorchTPU,其 Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,不再需要 torch_xla 或 Optimum TPU。 据 Hugging Face 介绍,在 TPU 虚拟机上安装只需一条命令 pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu,该索引提供的是 torch 的 CPU 版本;加载模型时传入 device_map="tpu" 即可。Hugging Face 称,在 TPU 上运行的测试中 96.5% 通过。

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

报道时间线

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

10月20日
  1. Hugging Face Blog精选
    Hugging Face 接入 TorchTPU,Transformers 与 Diffusers 可原生运行在 TPU 上

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

本事件热度走势

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

木叶资讯微信公众号二维码

关注公众号,每天收到精选与决策卡

长按识别二维码