《獨家新聞》谷歌擬提升TPU芯片與PyTorch的兼容性搶占AI計算市場

路透12月17日 - 據消息人士透露，Alphabet GOOGL.O旗下谷歌正在推進一項新計劃，以提升其人工智能(AI)芯片在運行全球最常用的 AI 軟件框架 PyTorch 時的性能，此舉旨在削弱Nvidia(輝達/英偉達) NVDA.O在 AI計算市場的長期主導地位。

該舉措是谷歌雄心勃勃的計劃的一部分，該計劃旨在推動其張量處理單元（TPU）成為Nvidia在市場上領先的圖形處理器（GPU）的有力競爭對手。TPU 銷售已成為谷歌雲收入的關鍵增長引擎，谷歌希望向投資者證明其 AI 投資正在產生回報。

但僅靠硬件不足以推動採用。消息人士稱，這項內部代號為“TorchTPU”的新計劃旨在消除阻礙 TPU 芯片普及的關鍵障礙，使其與 PyTorch 完全兼容，並為已使用 PyTorch 構建系統的客戶提供更友好的開發體驗。一些消息人士表示，谷歌還考慮將部分軟件開源，以加快客戶採用速度。

消息人士稱，與此前支持 PyTorch 在 TPU 上運行的嘗試相比，谷歌在 TorchTPU 上投入了更多組織資源和戰略重視，因為希望採用TPU芯片但認為軟件棧是瓶頸的公司的需求不斷增長。

PyTorch 是一個由 Meta Platforms META.O大力支持的開源項目，是開發 AI 模型的最常用工具之一。

Nvidia的工程師們多年來一直致力於確保使用 PyTorch 開發的軟件在其芯片上運行得盡可能快且高效。相比之下，谷歌長期以來一直讓其內部軟件開發團隊使用名為 Jax 的另一種開發框架，其 TPU 芯片則依賴名為 XLA 的工具來高效運行該代碼。谷歌自身的大部分 AI 軟件棧和性能優化都是圍繞 Jax 構建的，這進一步拉大了谷歌芯片與客戶需求之間的差距。

谷歌雲發言人沒有對該項目的具體細節發表評論，但向路透證實，此舉將為客戶提供更多選擇。

該發言人表示：“我們看到市場對 TPU 和 GPU 基礎設施的需求都在大幅加速增長。我們的重點是為開發者提供所需的靈活性和規模，無論他們選擇哪種硬件進行構建。”(完)