OpenAI今天宣布其未来的项目将迁移到Facebook的PyTorch机器学习框架,回避了谷歌日渐老迈的TensorFlow平台,这可以视作是这个社交网络巨头取得的胜利。
OpenAI是总部位于旧金山的AI研究公司,由首席技术官Greg Brockman、首席科学家Ilya Sutskever、Elon Musk及其他人共同创办,并得到LinkedIn联合创始人Reid Hoffman和Y Combinator前总裁Sam Altman等知名人士的支持。该公司在一篇博文中提到PyTorch的效率、可扩展性和采用率是决定改弦易辙的几个原因。
该公司在一份声明中说:“我们现统一使用PyTorch作为OpenAI的深度学习框架。在过去,我们根据各种框架的相对优点决定项目使用何种框架。现在我们决定统一采用PyTorch,以便我们的团队更容易创建和共享经过优化的模型。展望未来,我们将主要使用PyTorch作为我们的深度学习框架,但有时会因特定的技术原因而使用其他框架。我们很高兴能加入一个迅速壮大的开发者社区(包括Facebook和微软等企业组织),共同提升规模和性能。”
OpenAI表示,它的许多团队已经将工作迁移到PyTorch上,会在今后几个月为PyTorch社区做出贡献。此外该公司表示,计划在2020年初提供PyTorch方面的Spinning Up in Deep RL教学资源,此后打算研究利用数据并行训练扩展AI系统、利用模型可解释性直观显示那些系统,并建立通用的机器人框架。(OpenAI正在为其高度优化的blockparse内核编写PyTorch绑定代码,表示会在今后几个月内开源那些绑定代码。)
Facebook于2016年10月公开发布的PyTorch是一个基于Torch的开源机器学习库,Torch是一种科学计算框架和脚本语言,而这种语言又基于Lua编程语言。从2018年3月开始,PyTorch合并了Caffe2,这种深度学习工具集由加利福尼亚大学伯克利分校的研究人员率先开发,并由Facebook的AI研究实验室进一步开发。
虽然TensorFlow的历史要久一点(自2015年11月问世),但PyTorch继续在数据科学和开发者社区迅速受到追捧。据GitHub的2018年Octoverse报告显示,PyTorch在过去12个月成为增长速度最快的开源项目之一。Facebook最近透露,2019年,为该平台贡献代码的人数同比猛增50%以上,达到近1200人。The Gradient网站进行的分析发现,2019年的各大AI会议上展示的绝大多数论文提到的技术是用PyTorch实现的。O'Reilly特别指出,单单在2019年上半年,PyTorch在论文中的引用率剧增了194%以上。
许多领先的机器学习软件项目都建立在PyTorch的基础上,包括优步的Pyro和HuggingFace的Transformers,这不足为奇。软件开发商Preferred Networks最近加入了这个行列,承诺在不久的将来从其定制的AI开发框架Chainer迁移到PyTorch。
领取专属 10元无门槛券
私享最新 技术干货