OpenAI近日宣布可能會在將來把項目遷移到PyTorch機器學習框架,而不是在多種平臺間左右搖擺。OpenAI是位于舊金山的AI研究公司,由CTO Greg Brockman,首席科學家Ilya Sutskever,Elon Musk和其他人共同創立,并得到LinkedIn聯合創始人Reid Hoffman和前Y Combinator總裁Sam Altman等名人的支持。在一篇博客文章中,該公司將PyTorch的效率、可擴展性和采用率作為其決定的原因。

該公司在一份聲明中說:“展望未來,我們將主要使用PyTorch作為我們的深度學習框架,但有時在有特定技術原因時使用其他框架。我們很高興能加入一個快速增長的開發人員社區,包括Facebook和Microsoft這樣的組織,以推動圖形卡的規模和性能?!?/span>
OpenAI表示,其許多團隊已經將其工作遷移到PyTorch,并將在未來幾個月內為PyTorch社區做出貢獻。此外,該公司表示,計劃在2020年初在PyTorch?上提供Spinning Up in Deep RL教育資源,此后,該公司打算研究具有數據并行訓練的可擴展AI系統,以模型可解釋性可視化這些系統,并建立通用的目標機器人框架。(OpenAI正在為其高度優化的blockparse內核編寫PyTorch綁定,并表示將在未來幾個月內開源這些綁定。)
PyTorch于2016年10月在Facebook上公開發布,是一個基于Torch的開源機器學習庫,Torch是一種科學計算框架和腳本語言,而該腳本語言又是基于Lua編程語言的。截至2018年3月,它并入了Caffe2,這是由加利福尼亞大學伯克利分校的研究人員率先開發,并由Facebook的AI研究實驗室進一步開發的深度學習工具集。

盡管TensorFlow的使用時間略長一些(自2015年11月起),但PyTorch繼續在數據科學和開發人員社區中迅速普及。根據GitHub的2018 Octoverse報告,它在過去12個月中成為增長最快的開源項目平臺之一。Facebook最近透露,在2019年,該平臺的貢獻者數量同比增長了50%以上,達到近1200名。The Gradient進行的分析發現,PyTorch實施了2019年每個主要的AI會議,并且O’Reilly?指出,僅在2019年上半年,PyTorch對論文的引用增加了194%以上。
毫不奇怪,在PyTorch之上構建了許多領先的機器學習軟件項目,包括Uber的Pyro和HuggingFace的Transformers。軟件開發商Preferred Networks最近加入了這一行列,并承諾在不久的將來從其定制的AI開發框架Chainer遷移到PyTorch。