Hugging Face 宣布 TorchTPU 原生支持 TPU

Hugging Face 宣布推出 TorchTPU，作为 PyTorch 原生后端，使 Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行。用户只需指定 device_map="tpu" 或 pipe.to("tpu")，不再需要 torch_xla 或 Optimum TPU。 在 TPU 虚拟机上安装只需一条命令：pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu，该索引提供 torch 的 CPU 构建。TPU 可在 Google Cloud 上使用，从单芯片到数千芯片的 pod。Hugging Face 称正在为 TPU 添加 nightly CI，并将结果发布在 Transformers 仪表盘上。

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

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