Google 開源 TorchTPU，PyTorch 原生支援 TPU
先了解這件事AI 綜述
2026 年 10 月 20 日，Google 開源 TorchTPU，為 PyTorch 提供原生 TPU 後端，PyPI 最新版本為 0.2.0。據 Hugging Face 部落格，Transformers、Diffusers 與 Kernels 現在可直接在 TPU 上執行，模型只要送到 "tpu" 即可，不再需要 torch_xla 或 Optimum TPU，安裝只需一條 pip 命令。Hacker News 條目補充，專案包含面向 TPU 的 ATen 核心實現與 XLA 計算編譯快取，支援資料並行與張量並行分散式訓練，並接入 torch.compile() 圖模式執行，附帶 Llama、Qwen、ResNet、DLRM、minGPT、UNet 等範例；要求 Python 3.11+，Marimo 互動筆記本需在 TPU VM（如 v6e 例項）上執行。據 Google Developers 部落格，TorchTPU 把 torch.device("tpu") 接入 PyTorch，配套的 TorchTitan TPU 分支提供 SimpleFSDP、Graph Trainer 和自訂 Pallas 核心，在稠密與 MoE 架構上取得較高 MFU，官方還給出最佳化 playbook、recipes 與 Hugging Face 建模後端。對原先依賴 torch_xla 或被迫轉用 JAX 的 PyTorch 開發者來說，在 Google Cloud TPU 上訓練與部署模型的門檻降低了。
AI 根據 3 則報導生成 · 更新，有新報導，綜述更新中
報導時間線
沿著報導，了解事件的不同側面；點標題看單篇報導的摘要與原文連結。
沒有符合條件的報導。
本事件熱度走勢
為什麼熱
過去 48 小時，有 3 個獨立來源報導，最近 6 小時新增 2 個。熱度 2.4，熱門榜第 1 名。
熱度以 48 小時內的獨立來源計，同一來源只算一次，每 24 小時權重減半。
事件紀錄
- 最早報導
- 最近更新
同一事件的報導集中在這裡，新的進展會繼續補充。