Hugging Face 博客介绍 TorchTPU 原生 TPU 后端

Hugging Face 发布博客称，TorchTPU 为 PyTorch 新增原生 TPU 后端，TPU 由此成为 PyTorch 的普通设备；Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行，不再需要 torch_xla 或 Optimum TPU。 据该博客，在 TPU 虚拟机上安装只需一条命令 pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu，该额外索引提供 torch 的 CPU 构建；首次调用会编译，之后相同图像尺寸的调用复用已编译图。博客称在 TPU 上运行的测试中 96.5% 通过。

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

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