Google 携手 Meta 开源 TorchTPU,PyTorch 原生跑上 TPU
PyTorch · x · 2026-10-22
Google 与 Meta 官方宣布 TorchTPU 开源:为 Google TPU 提供原生 PyTorch 后端,现有 PyTorch 工作负载可开箱即用地迁移到 TPU。
- 由 Google 于 2025 年年中启动、与 Meta PyTorch 团队合作开发,在 PyTorch Conf 2026 发布
- 与旧方案 PyTorch/XLA(基于 Lazy Tensors、需侵入式改代码、编译延迟高)不同,TorchTPU 把 eager 执行作为首要目标,并集成 torch.compile 与原生硬件加速
- 兼容 vLLM、SGLang、TorchTitan、Hugging Face、Ray 等主流生态工具
这是 TPU 生态从 JAX/TensorFlow 向 PyTorch 主流社区开放的重要一步。
所属事件:谷歌联手 Meta 开源 TorchTPU，PyTorch 原生跑上 TPU(15 条相关)→
「Infra」频道最新
- GPU 新原语公司 Flappy 获 10 亿美元 A 轮融资，Index 与 Lux 联合领投 — gabriel1 · 2026-10-22
- 研究者断言:终极 megakernel 不是内核,而是无限循环的 Transformer — DimitrisPapail · 2026-10-22
- SPCX 财报电话会提出「每瓦算力」指标,AI 竞争下沉到底层资源 — mobileraj · 2026-10-22
- Eureka Machine 适配 Mac 推理,MacBook 满载几乎无风扇噪音 — cong_ml · 2026-10-22
- 算力需求过热变「逐月拍卖」，分析师称长期合约模式正在消亡 — BenBajarin · 2026-10-22
- SGLang 团队用 TPU + TorchTitan 跑早期 RL 训练，结果令人鼓舞 — ying11231 · 2026-10-22