據 AIbase 報道，為了撼動英偉達在 AI 芯片領域的統治地位，Alphabet（谷歌母公司）正在推進一項名為“TorchTPU”的戰略計劃。該計劃旨在顯著提升其張量處理單元（TPU）芯片對 PyTorch 框架的兼容性，從而降低開發者從英偉達 GPU 切換至谷歌 TPU 的技術門檻和遷移成本。
長期以來，PyTorch 作為全球應用最廣泛的開源 AI 開發框架，與英偉達的 CUDA 軟件棧深度綁定，這構成了英偉達最強大的生態護城河。相比之下，谷歌的 TPU 過去主要適配自家的 JAX 框架，導致大量習慣使用 PyTorch 的開發者在調用 TPU 算力時面臨性能瓶頸。通過 TorchTPU 項目，谷歌計劃投入更多戰略資源，優化底層軟件與 PyTorch 的匹配度。
谷歌正考慮開源部分核心軟件組件，以吸引更多開發者。據悉，谷歌還與 PyTorch 的主要維護方 meta 展開了深度合作，探討讓 meta 獲取更多 TPU 資源的可能性。在硬件層面，谷歌最新發布的第七代 TPU v7（代號 Ironwood） 已針對推理場景進行了大幅優化。
通過 TorchTPU 補齊軟件短板后，谷歌有望為企業提供一個更具性價比的英偉達替代方案，加速其 AI 基礎設施的商業化進程。
