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 也計算進去）。

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

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