目錄
TensorFlow的黃昏和PyTorch的崛起
Google機器學習的未來-JAX
Google的挑戰
首頁 科技週邊 人工智慧 被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

May 04, 2023 am 08:16 AM
Google tensorflow meta

很喜歡有些網友的一句話:

「這孩子實在不行,咱再要一個吧。」

Google還真這麼做了。

養了七年的TensorFlow終於還是被Meta的PyTorch幹趴下了,在某種程度上。

Google眼見不對,趕緊又要了一個--「JAX”,一款全新的機器學習框架。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

最近超級火辣的DALL·E Mini都知道吧,它的模型就是基於JAX進行程式設計的,從而充分地利用了GoogleTPU帶來的優勢。

TensorFlow的黃昏和PyTorch的崛起

2015年,Google開發的機器學習框架-TensorFlow問世。

當時,TensorFlow只是Google Brain的一個小專案。

誰也沒想到,剛一問世,TensorFlow就變得非常火爆。

優步、愛彼迎這種大公司在用,NASA這種國家機構也在用。而且還都是用在他們各自最複雜的項目上。

而截止到2020年11月,TensorFlow的下載次數已經達到了1.6億次。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

不過,Google好像並沒有十分在乎這麼多使用者的感受。

奇奇怪怪的介面和頻繁的更新都讓TensorFlow對使用者越來越不友好,而且越來越難以操作。

甚至,就連Google內部,也覺得這個框架在走下坡。

其實Google如此頻繁的更新也實屬無奈,畢竟只有這樣才能追得上機器學習領域快速地迭代。

於是,越來越多的人加入了這個項目,導致整個團隊慢慢失去了重點。

而原本讓TensorFlow成為首選工具的那些閃光點,也被埋沒在了茫茫多的要素裡,不再受人重視。

這種現像被Insider形容為一種「貓鼠遊戲」。公司就像是一隻貓,不斷迭代出現的新需求就像是一隻老鼠。貓要時時保持警惕,隨時撲向老鼠。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

這種困局對最先打入某一市場的公司來說是避不開的。

舉個例子,就搜尋引擎來說,Google並不是第一家。所以Google能夠從前輩(AltaVista、Yahoo等等)的失敗中總結經驗,應用在自身的發展上。

可惜到了TensorFlow這裡,Google是被困住的那一個。

正是因為上面這些原因,原先給Google賣命的開發者,慢慢對老東家失去了信心。

昔日無處不在的TensorFlow漸漸隕落,敗給了Meta的後起之秀——PyTorch。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

2017年,PyTorch的測試版開源。

2018年,Facebook的人工智慧研究實驗室發布了PyTorch的完整版本。

值得一提的是,PyTorch和TensorFlow都是基於Python開發的,而Meta則更注重維護開源社區,甚至不惜大量投入資源。

而且,Meta關注到了Google的問題所在,認為不能重蹈覆轍。他們專注於一小部分功能,並把這些功能做到最好。

Meta並沒有步上谷歌的後塵。這款首先在Facebook開發出來的框架,慢慢成為了業界標竿。

一家機器學習新創公司的研究工程師表示,「我們基本上都用PyTorch。它的社群和開源做得是最出色的。不僅有問必答,給的例子也很實用。」

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

面對這種局面,Google的開發者、硬體專家、雲端供應商,以及任何和Google機器學習相關的人員在接受採訪時都說了一樣的話,他們認為TensorFlow失掉了開發者的心。

經歷了一系列的明爭暗鬥,Meta最終佔了上風。

有專家表示,Google未來持續引領機器學習的機會正慢慢流失。

PyTorch逐漸成為了尋常開發者和研究人員的首選工具。

從Stack Overflow提供的互動資料來看,在開發者論壇上有關PyTorch的提問越來越多,而關於TensorFlow的最近幾年一直處於停滯狀態。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

就連文章開始提到的優步等等公司也轉向PyTorch了。

甚至,PyTorch後來的每一次更新,都像是在打TensorFlow的臉。

Google機器學習的未來-JAX

就在TensorFlow和PyTorch打得熱火朝天的時候,Google內部的一個「小型黑馬研究團隊」開始致力於開發一個全新的框架,可以更便捷地利用TPU。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

2018年,一篇題為《Compiling machine learning programs via high-level tracing》的論文,讓JAX專案浮出水面,作者是Roy Frostig、Matthew James Johnson和Chris Leary。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

從左至右依序是這三位大神

#而後,PyTorch原始作者之一的Adam Paszke,也在2020年初全職加入了JAX團隊。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

JAX提供了一個更直接的方法來處理機器學習中最複雜的問題之一:多核心處理器調度問題。

根據所應用的情況,JAX會自動地將若干個晶片組合而成一個小團體,而不是讓一個去單打獨鬥。

如此帶來的好處就是,讓盡可能多的TPU片刻間就能得到回應,從而燃燒我們的「煉丹小宇宙」。

最終,相較於臃腫的TensorFlow,JAX解決了Google內部的一個心頭大患:如何快速存取TPU。

下面簡單介紹一下構成JAX的Autograd和XLA。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

Autograd主要應用於基於梯度的最佳化,可以自動區分Python和Numpy程式碼。

它既可以用來處理Python的一個子集,包括循環、遞歸和閉包,也可以對導數的導數進行求導。

此外,Autograd支援梯度的反向傳播,這也就這意味著它可以有效地獲取標量值函數相對於數組值參數的梯度,以及前向模式微分,並且兩者可以任意組合。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

XLA(Accelerated Linear Algebra)可以加速TensorFlow模型而無需更改原始程式碼。

當一個程式執行時,所有的操作都由執行器單獨執行。每個操作都有一個預先編譯的GPU核心實現,執行器會分派到該核心實現。

舉個栗子:

<span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">def</span> <span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">model_fn</span>(<span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">x</span>, <span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">y</span>, <span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">z</span>):<br><span style="color: rgb(215, 58, 73); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">return</span> <span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">tf</span>.<span style="color: rgb(0, 92, 197); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">reduce_sum</span>(<span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">x</span> <span style="color: rgb(215, 58, 73); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">+</span> <span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">y</span> <span style="color: rgb(215, 58, 73); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">*</span> <span style="color: rgb(89, 89, 89); margin: 0px; padding: 0px; background: none 0% 0% / auto repeat scroll padding-box border-box rgba(0, 0, 0, 0);">z</span>)
登入後複製

在沒有XLA的情況下運行,該部分會啟動三個內核:一個用於乘法,一個用於加法,一個用於減法。

而XLA可以透過將加法、乘法和減法「融合」到單一GPU核心中,從而實現最佳化。

這種融合操作不會將由記憶體產生的中間值寫入y*z記憶體x y*z;相反,它將這些中間計算的結果直接「串流」給用戶,同時將它們完全保存在GPU中。

在實務上,XLA可以實現約7倍的效能改進和約5倍的batch大小改進。

此外,XLA和Autograd可以任意組合,甚至可以利用pmap方法一次使用多個GPU或TPU核心進行程式設計。

而將JAX與Autograd和Numpy結合的話,就可以獲得一個面向CPU、GPU和TPU的易於程式設計且高效能的機器學習系統了。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

顯然,Google這次吸取了教訓,除了在自家全面鋪平以外,在推進開源生態的建設方面,也是格外地積極。

2020年DeepMind正式投入JAX的懷抱,而這也宣告了谷歌親自下場,自此之後各種開源的庫層出不窮。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

「縱觀整場「明爭暗鬥」,賈揚清表示,在批評TensorFlow的過程中,AI系統認為Pythonic的科研就是全部需求。

但一方面純Python無法實現高效率的軟硬協同設計,另一方面上層分散式系統仍需要高效率的抽象。

而JAX正是在尋找更好的平衡,Google這種願意顛覆自己的pragmatism非常值得學習。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

causact R軟體包和相關貝葉斯分析教科書的作者表示,自己很高興看到Google從TF過渡到JAX,一個更乾淨的解決方案。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

Google的挑戰

作為新秀,Jax雖然可以藉鏡PyTorch和TensorFlow這兩位老前輩的優點,但有的時候後發可能也會帶來劣勢。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

首先,JAX還太「年輕」,作為實驗性的框架,遠遠沒有達到一個成熟的Google產品的標準。

除了各種隱藏的bug以外,JAX在某些問題上仍然要依賴其他框架。

拿載入和預處理資料來說,就需要用TensorFlow或PyTorch來處理大部分的設定。

顯然,這和理想的「一站式」框架還相去甚遠。

被PyTorch打爆! Google拋棄TensorFlow,押寶JAX

其次,JAX主要針對TPU進行了高度的最佳化,但是到了GPU和CPU上,就要差很多了。

一方面,Google在2018年至2021年組織和策略的混亂,導致在對GPU進行支援上的研發的資金不足,以及對相關問題的處理優先順序靠後。

同時,大概是過於專注於讓自家的TPU能在AI加速上分得更多的蛋糕,和英偉達的合作自然十分匱乏,更不用說完善對GPU的支持這種細節問題了。

另一方面,Google自己的內部研究,不用想肯定都集中在TPU上,這就導致Google失去了對GPU使用的良好回饋迴路。

此外,更長的偵錯時間、並未與Windows相容、未追蹤副作用的風險等等,都增加了Jax的使用門檻以及友善程度。

現在,PyTorch已經快6歲了,但完全沒有TensorFlow當年顯現出的頹勢。

如此看來,想要後來者居上的話,Jax還有很長一段路要走。


以上是被PyTorch打爆! Google拋棄TensorFlow,押寶JAX的詳細內容。更多資訊請關注PHP中文網其他相關文章!

本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn

熱AI工具

Undresser.AI Undress

Undresser.AI Undress

人工智慧驅動的應用程序,用於創建逼真的裸體照片

AI Clothes Remover

AI Clothes Remover

用於從照片中去除衣服的線上人工智慧工具。

Undress AI Tool

Undress AI Tool

免費脫衣圖片

Clothoff.io

Clothoff.io

AI脫衣器

Video Face Swap

Video Face Swap

使用我們完全免費的人工智慧換臉工具,輕鬆在任何影片中換臉!

熱工具

記事本++7.3.1

記事本++7.3.1

好用且免費的程式碼編輯器

SublimeText3漢化版

SublimeText3漢化版

中文版,非常好用

禪工作室 13.0.1

禪工作室 13.0.1

強大的PHP整合開發環境

Dreamweaver CS6

Dreamweaver CS6

視覺化網頁開發工具

SublimeText3 Mac版

SublimeText3 Mac版

神級程式碼編輯軟體(SublimeText3)

熱門話題

Java教學
1653
14
CakePHP 教程
1413
52
Laravel 教程
1306
25
PHP教程
1251
29
C# 教程
1224
24
芝麻開門交易所網頁版登入口 最新版gateio官網入口 芝麻開門交易所網頁版登入口 最新版gateio官網入口 Mar 04, 2025 pm 11:48 PM

詳細介紹芝麻開門交易所網頁版登入口操作,含登錄步驟、找回密碼流程,還針對登錄失敗、無法打開頁面、收不到驗證碼等常見問題提供解決方法,助你順利登錄平台。

芝麻開門交易所網頁註冊鏈接 gate交易app註冊網址最新 芝麻開門交易所網頁註冊鏈接 gate交易app註冊網址最新 Feb 28, 2025 am 11:06 AM

本文詳細介紹了芝麻開門交易所(Gate.io)網頁版和Gate交易App的註冊流程。 無論是網頁註冊還是App註冊,都需要訪問官方網站或應用商店下載正版App,然後填寫用戶名、密碼、郵箱和手機號等信息,並完成郵箱或手機驗證。

芝麻開門交易平台下載手機版 gateio交易平台下載地址 芝麻開門交易平台下載手機版 gateio交易平台下載地址 Feb 28, 2025 am 10:51 AM

選擇正規渠道下載App,保障您的賬戶安全至關重要。

加密數字資產交易APP推薦top10(2025全球排名) 加密數字資產交易APP推薦top10(2025全球排名) Mar 18, 2025 pm 12:15 PM

本文推荐十大值得关注的加密货币交易平台,涵盖币安(Binance)、OKX、Gate.io、BitFlyer、KuCoin、Bybit、Coinbase Pro、Kraken、BYDFi和XBIT去中心化交易所。这些平台在交易币种数量、交易类型、安全性、合规性、特色功能等方面各有千秋,例如币安以其全球最大的交易量和丰富的功能著称,而BitFlyer则凭借其日本金融厅牌照和高安全性吸引亚洲用户。选择合适的平台需要根据自身交易经验、风险承受能力和投资偏好进行综合考量。 希望本文能帮助您找到最适合自

歐易okex賬號怎麼註冊、使用、註銷教程 歐易okex賬號怎麼註冊、使用、註銷教程 Mar 31, 2025 pm 04:21 PM

本文詳細介紹了歐易OKEx賬號的註冊、使用和註銷流程。註冊需下載APP,輸入手機號或郵箱註冊,完成實名認證。使用方面涵蓋登錄、充值提現、交易以及安全設置等操作步驟。而註銷賬號則需要聯繫歐易OKEx客服,提供必要信息並等待處理,最終獲得賬號註銷確認。 通過本文,用戶可以輕鬆掌握歐易OKEx賬號的完整生命週期管理,安全便捷地進行數字資產交易。

如何在Bitget官網註冊並下載最新App 如何在Bitget官網註冊並下載最新App Mar 05, 2025 am 07:54 AM

本指南提供了 Bitget 交易所官方 App 的詳細下載和安裝步驟,適用於安卓和 iOS 系統。指南整合了來自多個權威來源的信息,包括官網、App Store 和 Google Play,並強調了下載和賬戶管理過程中的注意事項。用戶可以從官方渠道下載 App,包括應用商店、官網 APK 下載和官網跳轉,並完成註冊、身份驗證和安全設置。此外,指南還涵蓋了常見問題和注意事項,例如

為什麼說Bittensor是AI賽道的'比特幣”? 為什麼說Bittensor是AI賽道的'比特幣”? Mar 04, 2025 pm 04:06 PM

原文標題:Bittensor=AIBitcoin?原文作者:S4mmyEth,DecentralizedAIResearch原文編譯:zhouzhou,BlockBeats編者按:本文討論了Bittensor,一個去中心化的AI平台,希望通過區塊鏈技術打破集中式AI公司的壟斷,推動開放、協作的AI生態系統。 Bittensor採用子網模型,允許不同AI解決方案的出現,並通過TAO代幣激勵創新。儘管AI市場已成熟,但Bittensor面臨競爭風險,可能會受到其他開源

binance怎麼註冊詳細教程(2025新手指南) binance怎麼註冊詳細教程(2025新手指南) Mar 18, 2025 pm 01:57 PM

本文提供Binance幣安註冊及安全設置的完整指南,涵蓋註冊前的準備工作(包括設備、郵箱、手機號及身份證明文件準備),詳細介紹了官網及APP兩種註冊方式,以及不同級別的身份驗證(KYC)流程。此外,文章還重點講解瞭如何設置資金密碼、開啟雙重驗證(2FA,包括谷歌身份驗證器和短信驗證)以及設置防釣魚碼等關鍵安全步驟,幫助用戶安全便捷地註冊和使用Binance幣安平台進行加密貨幣交易。 請務必在交易前了解相關法律法規及市場風險,謹慎投資。

See all articles