Hugging Face 发布 TorchTPU 原生 PyTorch TPU 后端

Hugging Face 发布 TorchTPU，为 PyTorch 增加 TPU 原生后端，使 TPU 成为 PyTorch 的普通设备。据其博客，Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行，模型通过 device_map="tpu" 或 pipe.to("tpu") 加载。 在 TPU VM 上安装只需一条命令：pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu；该额外索引提供的是 torch 的 CPU 构建。Hugging Face 称，在 TPU 上运行的测试中 96.5% 通过，并正着手添加 TPU 上的 nightly CI，将结果发布到 Transformers 仪表盘。

AI 根据报道生成 · 13 天前更新