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

Hugging Face 生态原生支持 TPU,Google 与 Meta 发布 TorchTPU

2 篇报道2 个报道来源2 小时前更新

先了解这件事

AI 综述

Google 与 Meta 发布开源项目 TorchTPU,为 Google TPU 提供原生 PyTorch 后端;据 PyTorch 博客,模型迁移只需把 .to("cuda") 改为 .to("tpu")。 Hugging Face 随后宣布,其 Transformers、Diffusers 和 Kernels 已通过 TorchTPU 原生运行 TPU,模型加载只需 device_map="tpu"。在 TPU 虚拟机上安装 TorchTPU 只需一条命令:pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu。 Hugging Face 称,TorchTPU 让 TPU 成为 PyTorch 的又一种普通设备,因此用 Transformers 和 Diffusers 运行 Hugging Face 模型已不再需要 Optimum TPU。

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

事件进展

2 个进展

  1. 10月21日 00:00 · 1 篇报道
    TorchTPU开源发布
    PyTorch:Blog:Google 与 Meta 发布 TorchTPU:PyTorch 原生 TPU 后端
  2. 10月20日 08:00 · 1 篇报道
    Hugging Face 宣布 TorchTPU 原生支持
    Hugging Face 官方团队博客:Hugging Face 生态通过 TorchTPU 原生支持 TPU

报道时间线

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

10月21日
  1. PyTorch:Blog精选
    Google 与 Meta 发布 TorchTPU:PyTorch 原生 TPU 后端

    Google 与 Meta 发布开源项目 TorchTPU,为 Google TPU 提供原生 PyTorch 后端,模型迁移只需把 .to("cuda") 改为 .to("tpu")。

10月20日
  1. Hugging Face 官方团队博客
    Hugging Face 生态通过 TorchTPU 原生支持 TPU

    Hugging Face 宣布通过 PyTorch 原生后端 TorchTPU 支持 TPU,Transformers、Diffusers 和 Kernels 可原生运行,模型加载只需 device_map="tpu",不再需要 torch_xla 或 Optimum TPU。

本事件热度走势

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