Karpathy更新深度学习开源框架排名：TensorFlow第一，PyTorch第二

Posted 2021-05-02 新智元

【2018 新智元 AI 技术峰会倒计时 17 天】

3月29日，将于北京举办的 2018 年中国 AI 开年盛典——2018 新智元 AI 技术峰会，我们邀请到了德国总理默克尔的科学顾问、诺贝尔奖唯一计算机领域评委、工业 4.0 教父、世界顶级自然语言处理专家 Wolfgang Wahlste 教授。Wahlster 教授将亲临 329 峰会现场，分享欧洲对人工智能科技发展和 AI 产业化的思考。与诺奖评委面对面，点击文末阅读原文，马上参会！

 

抢票链接：

http://www.huodongxing.com/event/8426451122400

---

  新智元报道  

【新智元导读】上周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

我们期待最新一度的统计结果。

P.S. 真的没有找到PaddlePaddle。

了解更多：Andrej Karpathy从Arxiv论文看机器学习趋势：

https://medium.com/@karpathy/a-peek-at-trends-in-machine-learning-ab8a1085a106

---

【2018 新智元 AI 技术峰会倒计时 17 天】

点击阅读原文查看嘉宾与日程

 

 峰会门票火热抢购，抢票链接：

http://www.huodongxing.com/event/8426451122400

 

【扫一扫或点击阅读原文抢购大会门票】