據(jù) AIbase 報(bào)道，為了撼動(dòng)英偉達(dá)在 AI 芯片領(lǐng)域的統(tǒng)治地位，Alphabet（谷歌母公司）正在推進(jìn)一項(xiàng)名為“TorchTPU”的戰(zhàn)略計(jì)劃。該計(jì)劃旨在顯著提升其張量處理單元（TPU）芯片對(duì) PyTorch 框架的兼容性，從而降低開發(fā)者從英偉達(dá) GPU 切換至谷歌 TPU 的技術(shù)門檻和遷移成本。

長期以來，PyTorch 作為全球應(yīng)用最廣泛的開源 AI 開發(fā)框架，與英偉達(dá)的 CUDA 軟件棧深度綁定，這構(gòu)成了英偉達(dá)最強(qiáng)大的生態(tài)護(hù)城河。相比之下，谷歌的 TPU 過去主要適配自家的 JAX 框架，導(dǎo)致大量習(xí)慣使用 PyTorch 的開發(fā)者在調(diào)用 TPU 算力時(shí)面臨性能瓶頸。通過 TorchTPU 項(xiàng)目，谷歌計(jì)劃投入更多戰(zhàn)略資源，優(yōu)化底層軟件與 PyTorch 的匹配度。

谷歌正考慮開源部分核心軟件組件，以吸引更多開發(fā)者。據(jù)悉，谷歌還與 PyTorch 的主要維護(hù)方 meta 展開了深度合作，探討讓 meta 獲取更多 TPU 資源的可能性。在硬件層面，谷歌最新發(fā)布的第七代 TPU v7（代號(hào) Ironwood） 已針對(duì)推理場(chǎng)景進(jìn)行了大幅優(yōu)化。

通過 TorchTPU 補(bǔ)齊軟件短板后，谷歌有望為企業(yè)提供一個(gè)更具性價(jià)比的英偉達(dá)替代方案，加速其 AI 基礎(chǔ)設(shè)施的商業(yè)化進(jìn)程。