AISpot
熱點事件 · 持續更新

Google 開源 TorchTPU,PyTorch 原生支援 TPU

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

先了解這件事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. TechCrunch AI● 精選

    Google 開源 TorchTPU,PyTorch 應用改配置即可跑在 TPU 上

    Google 宣布開源 TorchTPU,PyTorch 應用只需改配置、無需重寫程式碼,就能執行在 Google 的 TPU 與 Google Cloud 上。據稱 85% 的 AI 應用基於 PyTorch,而它的許多特性原本專為 Nvidia GPU 設計。前一天 Amazon 也發布 TorchNeuron 支援自家 Trainium,並向 PyTorch 主專案提交程式碼,把各項特性與 Nvidia 晶片解耦。

  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.4,熱門榜第 1 名。

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

事件紀錄

最早報導
最近更新

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