Hugging Face 原生支持 TorchTPU，TPU 可直接运行

Hugging Face 宣布将 TorchTPU 作为 PyTorch 原生后端接入其生态，Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行，不再需要 Optimum TPU。 在 TPU 虚拟机上安装需执行一条命令：pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu。Hugging Face 称，该额外索引提供 torch 的 CPU 版本；若不使用，pip 会拉取体积更大的 CUDA 版本，而 TPU 上并不需要。加载模型时传入 device_map="tpu" 即可。 Hugging Face 称，在 TPU 上运行的测试中，96.5% 通过。

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

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