曠視開(kāi)源其AI生產(chǎn)力平臺(tái)Brain++的核心組件——天元

3月25日，通過(guò)線上發(fā)布會(huì)，國(guó)內(nèi)人工智能公司曠視科技（下稱(chēng)“曠視”）正式對(duì)外宣布，開(kāi)源其AI生產(chǎn)力平臺(tái)Brain++的核心組件——天元（MegEngine）。

天元，意為圍棋棋盤(pán)中心點(diǎn)。曠視表示，希望越來(lái)越多的開(kāi)發(fā)者基于天元可以從0到1創(chuàng)造屬于自己的算法，在AI時(shí)代提供一套人人可用的生產(chǎn)力工具。

曠視聯(lián)合創(chuàng)始人、CTO唐文斌在發(fā)布會(huì)上介紹本次曠視天元共開(kāi)源約35萬(wàn)行代碼，包括 C++、CUDA和Python的代碼。

那么如何來(lái)理解Brain++，以及天元的作用呢？在當(dāng)天的發(fā)布會(huì)上，曠視聯(lián)合創(chuàng)始人、首席技術(shù)官唐文斌用廚房做了一個(gè)比喻。

“Brain++平臺(tái)的意義是為了能像Visual Studio一樣為用戶提供更完善的集成開(kāi)發(fā)環(huán)境，滿足AI開(kāi)發(fā)者從 AI 生產(chǎn)（輸出算法模型）到應(yīng)用（實(shí)現(xiàn)算法工程化封裝）各環(huán)節(jié)中，一站式、全流程算法研發(fā)的切實(shí)需求，將產(chǎn)品從實(shí)驗(yàn)室原型到工業(yè)部署原本數(shù)周或數(shù)月的時(shí)間成本，縮短到小時(shí)級(jí)?！碧莆谋笳f(shuō)。

簡(jiǎn)單來(lái)講，算法研發(fā)過(guò)程就像炒菜。在“美味算法”的研發(fā)中，數(shù)據(jù)就是各種各樣的食材，需要清洗，分類(lèi)管理，是“炒制”算法的原材料；而算法的訓(xùn)練和推理就像是烹制的過(guò)程，需要鍋具（深度學(xué)習(xí)框架）來(lái)承載；算力則是一灶猛火，火候到位才能燒得好菜。

如果沒(méi)有一套廚具和廚房，大廚就像在野外生存，不僅要逐兔追雞，還得自己打鐵鑄鍋、生火劈柴，菜品生產(chǎn)力很低。同樣的，對(duì)于有一定的機(jī)器學(xué)習(xí)經(jīng)驗(yàn)和框架基礎(chǔ)的開(kāi)發(fā)者來(lái)說(shuō)，他們的開(kāi)發(fā)工具往往需要東拼西湊，再?gòu)?qiáng)的學(xué)術(shù)造詣也無(wú)法加快算法的生產(chǎn)。

在傳統(tǒng)開(kāi)發(fā)環(huán)境中，AI開(kāi)發(fā)者們會(huì)根據(jù)方向不同選用不同的框架，但是從原型設(shè)計(jì)到部署還需要切換開(kāi)發(fā)工具，就像廚師每炒一道菜還要換個(gè)鍋，鍋和灶眼也不見(jiàn)得適配。

天元對(duì)于AI開(kāi)發(fā)者來(lái)說(shuō)，就是一套從菜譜到美味的端到端一體化智能鍋，覆蓋從訓(xùn)練到推理再到部署裝盤(pán)全流程。不僅自帶菜譜（模型庫(kù)）還支持自定義菜譜（模型導(dǎo)入），具備顯示屏靈活可控、方便操作，特別適合既要求美味，又追求效率和體驗(yàn)的 AI 生產(chǎn)者。

理解完深度學(xué)習(xí)框架的作用后，你或許會(huì)問(wèn)，曠視為什么要在這個(gè)時(shí)候選擇開(kāi)源？實(shí)際上，要將自己的代碼進(jìn)行開(kāi)源，并不是一件簡(jiǎn)單的事情。因?yàn)檫@會(huì)關(guān)系到框架技術(shù)是否成熟，生態(tài)環(huán)境是否發(fā)展完善，以及未來(lái)發(fā)展的方向。

據(jù)了解，天元從2014年開(kāi)始研發(fā)，2015年全員使用，到今年3月開(kāi)源，曠視耗費(fèi)資源打造優(yōu)化了6年，迭代到看8.0版本。曠視目前所有的算法都是基于天元 MegEngine 這個(gè)框架訓(xùn)練和推理的。

發(fā)布會(huì)上，天元開(kāi)源項(xiàng)目的產(chǎn)品負(fù)責(zé)人田忠博表示，天元是一套訓(xùn)練推理一體化、動(dòng)靜態(tài)合一的工業(yè)級(jí)深度學(xué)習(xí)框架，架構(gòu)上具體分為計(jì)算接口、圖表示、優(yōu)化與編譯、運(yùn)行時(shí)管理和計(jì)算內(nèi)核五層。

值得一提的是，天元的開(kāi)源也意味著，這個(gè)從中國(guó)本土AI企業(yè)走出來(lái)的深度學(xué)習(xí)框架將躋身全球主流框架。在天元之前，深度學(xué)習(xí)框架開(kāi)源的不在少數(shù)，比如從學(xué)術(shù)界走出的Caffe、 Torch 和 Theano，以及現(xiàn)如今引領(lǐng)產(chǎn)業(yè)界的TensorFlow。此外，其他科技公司也在積極開(kāi)發(fā)自己的深度學(xué)習(xí)框架，例如亞馬遜的MXNet，F(xiàn)acebook打造的PyTorch，Microsoft內(nèi)部開(kāi)源的CNTK，以及相對(duì)小眾的深度學(xué)習(xí)引擎DSSTNE等。