據知情人士透露，字母表公司旗下的谷歌正推進一項新計劃，旨在優化其人工智能芯片對全球最主流 AI 軟件框架 PyTorch 的運行能力。此舉劍指英偉達在人工智能計算市場長期以來的主導地位。
該計劃是谷歌雄心勃勃的戰略布局之一，目標是讓其張量處理器（TPU）成為英偉達市場領先的圖形處理器（GPU）的可靠替代品。隨著谷歌努力向投資者證明其人工智能投資已產生回報，張量處理器的銷售額已成為谷歌云業務收入的關鍵增長引擎。
但知情人士稱，僅靠硬件不足以推動市場普及。這項內部代號為 “TorchTPU” 的新計劃，核心目標是消除阻礙張量處理器普及的關鍵壁壘 —— 通過實現張量處理器與 PyTorch 軟件的完全兼容，并提升開發者使用友好度，滿足那些已基于 PyTorch 搭建技術架構的客戶需求。部分知情人士還表示，谷歌正考慮將該軟件的部分組件開源，以加快客戶的采用進程。
知情人士指出，相較于此前為支持張量處理器運行 PyTorch 所做的嘗試，谷歌此次對 TorchTPU 投入了更多的組織關注度、資源與戰略權重。原因在于，越來越多有意采用張量處理器的企業，都將軟件生態視為技術落地的瓶頸。
PyTorch 是一個由元宇宙平臺公司（Meta Platforms）深度支持的開源項目，也是開發者構建人工智能模型最常用的工具之一。在硅谷，極少有開發者會逐行編寫能在英偉達、超威半導體或谷歌芯片上運行的代碼。
相反，開發者們會依賴 PyTorch 這類工具 —— 它是一套預制代碼庫與框架的集合，可自動完成人工智能軟件開發中的諸多通用任務。PyTorch 于 2016 年首次發布，其發展歷程與英偉達的 CUDA 軟件緊密相連。部分華爾街分析師認為，CUDA 正是英偉達抵御競爭對手最強有力的 “護城河”。
英偉達的工程師耗費多年時間，確保基于 PyTorch 開發的軟件能在其芯片上實現極速且高效的運行。相比之下，谷歌長期以來一直讓內部龐大的軟件開發團隊使用另一款名為 Jax 的代碼框架，其張量處理器則通過 XLA 工具保障該代碼的高效運行。谷歌自身的大部分人工智能軟件生態與性能優化工作，均圍繞 Jax 展開，這就導致谷歌芯片的實際應用方式，與客戶的使用需求之間形成了巨大鴻溝。
谷歌云發言人未就該項目的具體細節置評，但向路透社證實，此舉將為客戶提供更多選擇。
該發言人表示：“我們看到市場對張量處理器與圖形處理器基礎設施的需求正在激增，且增速持續加快。我們的核心關注點是，無論開發者選擇基于何種硬件開展構建工作，都能為他們提供所需的靈活性與規模支持。”
面向客戶開放張量處理器
長期以來，字母表公司將其自研芯片（即張量處理器）的絕大部分產能都預留作內部使用。這一局面在 2022 年被打破 —— 當時谷歌云計算部門成功游說公司，獲得了張量處理器銷售業務的主導權。這一調整大幅提升了谷歌云的張量處理器配額。隨著客戶對人工智能技術的興趣日益濃厚，谷歌也在通過擴大張量處理器的產能與對外銷售量，把握市場機遇。
然而，全球多數人工智能開發者使用的 PyTorch 框架，與谷歌芯片當前深度優化的 Jax 框架并不兼容。這意味著，大多數開發者若想采用谷歌芯片，并使其性能媲美英偉達產品，就必須投入大量額外的工程開發工作。在日新月異的人工智能賽道上，這類工作需要耗費大量的時間與資金成本。
若谷歌的 “TorchTPU” 計劃能夠成功落地，將大幅降低那些尋求英偉達 GPU 替代品的企業的技術遷移成本。英偉達的市場主導地位，不僅源于其硬件優勢，更得益于其 CUDA 軟件生態 —— 該生態已深度嵌入 PyTorch，成為企業訓練和運行大型人工智能模型的默認方案。
知情人士稱，企業客戶一直向谷歌反饋，張量處理器在人工智能工作負載場景中較難落地，原因是這類芯片歷來要求開發者切換至谷歌內部青睞的機器學習框架 Jax，而非大多數人工智能開發者早已熟練使用的 PyTorch。
與元宇宙平臺公司聯手攻關
知情人士透露，為加快研發進度，谷歌正與 PyTorch 的開發及維護方元宇宙平臺公司展開緊密合作。此前《信息報》曾率先報道，兩家科技巨頭正在磋商相關合作協議，內容包括元宇宙平臺公司獲得更多張量處理器的使用權限。
谷歌最初為元宇宙平臺公司提供的服務，是由谷歌全權管理的模式：像元宇宙平臺這樣的客戶，需部署谷歌設計的芯片來運行谷歌的軟件與模型，同時由谷歌提供運維支持。知情人士表示，元宇宙平臺公司在推動張量處理器適配軟件研發方面有著明確的戰略考量 —— 此舉有助于降低其模型推理成本，推動自身人工智能基礎設施擺脫對英偉達 GPU 的依賴，從而增強在供應鏈談判中的話語權。
元宇宙平臺公司對此不予置評。
今年以來，谷歌已開始直接向客戶的數據中心銷售張量處理器，不再將其使用權限局限于自家云平臺。本月，谷歌資深高管阿明?瓦赫達特被任命為人工智能基礎設施業務負責人，直接向谷歌首席執行官桑達爾?皮查伊匯報。
谷歌需要這套基礎設施來支撐兩方面的需求：一是運行自身的人工智能產品，包括 Gemini 聊天機器人與人工智能驅動的搜索服務；二是為谷歌云的客戶提供算力支持 —— 目前谷歌云正面向安索普（Anthropic）等企業銷售張量處理器的使用權限。