TorchTPU 发布：PyTorch 原生支持 TPU

Hugging Face 宣布 TorchTPU 为 PyTorch 增加原生 TPU 后端，Transformers、Diffusers 和 Kernels 无需 torch_xla 或 Optimum TPU 即可在 TPU 上运行，加载 Hub 模型只需将 device_map 设为 tpu。 在 TPU VM 上可通过一条命令安装（pip install torch-tpu --extra-index-url https://download.pytorch.org/whl/cpu），该额外索引提供 torch 的 CPU 构建。Hugging Face 称首次调用会编译计算图，之后相同图像尺寸的调用可直接复用。

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

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