Karpathy 更新深度學習開源框架排名：TensorFlow 第一，PyTorch 第二

上週 Keras 作者 François Chollet 才曬了使用谷歌搜尋 Index 得到的開源深度學習框架排名，前天好久沒出山的大神 Andrej Karpathy 丟擲更新，使用更精確的資料，得到過去 6 年、4300 篇 ArXiv 機器學習論文裡提到深度學習框架：PyTorch 漲勢驚人。

上週，Keras 作者、谷歌研究科學家 François Chollet 曬出一張圖，他使用 Google Search Index，展示了過去三個月，ArXiv 上提到的深度學習框架排行，新智元也做了

報道

：

TensorFlow 排名第一，Keras 排名第二，之後依次是 Caffe、PyTorch、Theano，MXNet、Chainer，以及 CNTK。

不少人評論，咦，PyTorch 這麼靠後？CNTK 更是不科學……

昨天，手握 ArXiv Sanity 大資料、現在主管特斯拉人工智慧部門的 Andrej Karpathy，給出了更加精確的結果。

過去一個月，各個框架在論文中被提到（單次計算）的比例分別是：

TensorFlow 14.3%，PyTorch 4.7％，Keras 4.0％，Caffe 3.8％，Theano 2.3％，Torch 1.5％，MXNet、Chainer 和 CNTK 均小於 1％

：

上圖中，綠色是 Theano，紫紅色是 TensorFlow（上升最高的那條），藍色是 Keras，棕黃色是 Caffe，紫色是 Torch，紫紅色是 PyTorch，淺綠是 MXNet，藍色是 Chainer，紅色是 CNTK。

沒錯，上面的資料確實是

過去一個月

，Karpathy 也在後面給出了更正說明。

Karpathy 表示，綜合過去 6 年發表在 ArXiv 的 4300 篇機器學習論文（資料來源：cs。［CV|CL|LG|AI|NE］/stat。ML），根據其中各框架被提及的次數得到的總排名是：

TensorFlow 5.9%，Caffe 5.4%，Theano 3.2%，Keras 2.3%，Torch 1.6％，PyTorch 1%，其他 0.5%

Theano 餘威仍在，PyTorch 漲勢驚人。

要知道，在 2017 年 3 月，Karpathy 第一次做這個全面統計的時候，各框架的排名是：

那時候 PyTorch 才剛剛釋出不久，Theano 也沒有說退休。

一年過去，如果有什麼沒有變，那就是王者 TensorFlow——2017 年 3 月提交的所有論文中，有 10％提到了 TensorFlow。 當然，並非每篇論文都會宣告所使用的框架，但如果假設論文以某種固定的隨機機率獨立於框架來宣告框架，那麼機器學習社群裡看起來大約有 40％的人都在使用 TensorFlow（或者更多，如果把 Keras 也計算進去）。

2017 年 3 月的資料：紫色是 Caffe，綠色是 TensorFlow，紅色是 Keras，黃色是 Torch，淺藍色是 PyTorch，藍色是 Theano

我們期待最新一度的統計結果。

（本文首發於新智元微信公眾號：AI_era，歡迎關注。）