Hugging Face 接入 TorchTPU，原生支持 TPU

Hugging Face 宣布接入原生 PyTorch 后端 TorchTPU，其 Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行，不再需要 torch_xla 或 Optimum TPU。 在 TPU 虚拟机上安装只需一条命令：pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu。Hugging Face 称，加上该索引可获取 CPU 版 torch，否则 pip 会拉取体积更大的 CUDA 版。首次调用会编译，之后相同图像尺寸的调用复用计算图。 Hugging Face 称，在 TPU 上运行的测试中 96.5% 通过。

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

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