跳到正文
热点事件历史事件

Google 与 Meta 开源 TorchTPU,Hugging Face 原生支持 TPU

5 篇报道5 个报道来源13 天前更新

先了解这件事

报道摘要

Google 与 Meta 联合开源 TorchTPU,为 Google TPU 提供原生 PyTorch 后端,模型只需把 .to("cuda") 改成 .to("tpu") 即可开箱运行,无需 JAX 依赖。

摘自 PyTorch:Blog

报道时间线

沿着报道,了解事件的不同侧面。

10月21日
  1. Rohan Paul
    Google 开源 TorchTPU,为 TPU 提供原生 PyTorch 后端

    Google 开源 TorchTPU,为 TPU 提供原生 PyTorch 后端,替代此前的 PyTorch/XLA 层。它按写入顺序逐个运行算子以便像 GPU 一样逐步调试,再切换 torch.compile 获得生产速度;训练 Qwen 3 8B 时最高用到 52% 的峰值 TPU 算力。原文还提到 Amazon 也提供原生 PyTorch 后端,让代码脱离 Nvidia GPU 运行。

  2. TechCrunch:AI
    Google 开源 TorchTPU,与 Amazon TorchNeuron 争夺 Nvidia 的 PyTorch 阵地

    Google 于周二宣布其 PyTorch 版本 TorchTPU 以开源项目形式发布,让现有 PyTorch 应用只需简单配置改动即可运行在 Google TPU 和 Google Cloud 上。

  3. PyTorch:Blog精选
    Google 与 Meta 发布 TorchTPU:PyTorch 原生 TPU 后端

    Google 与 Meta 联合开源 TorchTPU,为 Google TPU 提供原生 PyTorch 后端,模型只需把 .to("cuda") 改成 .to("tpu") 即可开箱运行,无需 JAX 依赖。

10月20日
  1. MarkTechPost
    Google 开源 TorchTPU:PyTorch 原生 Cloud TPU 后端,与 Meta 联合开发

    Google 与 Meta 的 PyTorch 团队联合开发并在 PyTorch Conference 2026 上宣布开源 TorchTPU,让 PyTorch 通过 torch.device("tpu") 原生运行在 Cloud TPU 上,官方称其取代 PyTorch/XLA 并推荐用于所有新工作负载。

  2. Hugging Face 官方团队博客
    Hugging Face 生态通过 TorchTPU 原生支持 TPU

    Hugging Face 宣布通过 PyTorch 原生后端 TorchTPU 支持 TPU,Transformers、Diffusers 和 Kernels 可原生运行,模型加载只需 device_map="tpu",不再需要 torch_xla 或 Optimum TPU。