AISpot
熱點事件 · 持續更新

Google 開源 TorchTPU,PyTorch 原生支援 TPU

3 則報導3 個報導來源 更新

先了解這件事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 則報導生成 · 更新

報導時間線

沿著報導,了解事件的不同側面;點標題看單篇報導的摘要與原文連結。

10月20日星期二 · 3 則

  1. Google Developers blog

    TorchTPU 把原生 TPU 後端接入 PyTorch

    TorchTPU 將原生 torch.device("tpu") 後端接入 PyTorch,開發者無需改用 JAX 即可在 Google Cloud TPU 上訓練大模型。配套的 TorchTitan TPU 分支提供 SimpleFSDP、Graph Trainer 和自訂 Pallas 核心等生產級擴充套件,在稠密與 MoE 架構上取得較高 MFU。官方還給出最佳化 playbook、recipes 與 Hugging Face 建模後端,便於在 TPU 上擴充套件自有負載。

  2. Hacker News front page● 精選

    Google 開源 TorchTPU,為 TPU 提供 PyTorch 原生後端

    Google 開源 TorchTPU,讓 PyTorch 模型可直接跑在 TPU 上,PyPI 最新版本為 0.2.0。專案提供 TPU 的 ATen 核心實現、XLA 計算編譯快取,支援資料並行與張量並行分散式訓練,並接入 torch.compile() 圖模式執行,附帶 Llama、Qwen、ResNet、DLRM、minGPT、UNet 等範例。安裝要求 Python 3.11+,Marimo 互動筆記本需在 TPU VM(如 v6e 例項)上執行。

本事件熱度走勢

最高 2.6

為什麼熱

過去 48 小時,有 3 個獨立來源報導,最近 6 小時新增 2 個。熱度 2.5,熱門榜第 1 名。

熱度以 48 小時內的獨立來源計,同一來源只算一次,每 24 小時權重減半。

事件紀錄

最早報導
最近更新

同一事件的報導集中在這裡,新的進展會繼續補充。