曠視宣布開(kāi)源深度學(xué)習(xí)框架天元 縮短AI算法訓(xùn)練周期

3月25日下午消息，曠視通過(guò)線(xiàn)上發(fā)布會(huì)正式宣布開(kāi)源新一代AI生產(chǎn)力平臺(tái)Brain++的核心深度學(xué)習(xí)框架曠視天元（MegEngine）。

天元的前身是曠視內(nèi)部自用的算法訓(xùn)練推理引擎MegEngine，由曠視3名實(shí)習(xí)生在2013年啟動(dòng)研發(fā)，并與2014年正式完成上線(xiàn)，支撐了曠視6年以來(lái)在國(guó)際AI競(jìng)賽的表現(xiàn)，以及公司產(chǎn)品和業(yè)務(wù)的落地，目前服務(wù)于曠視研究院1400余名AI開(kāi)發(fā)者。

Brain++是曠視自主研發(fā)的端到端人工智能算法平臺(tái)，具備大規(guī)模算法研發(fā)能力。針對(duì)框架、算力和數(shù)據(jù)三個(gè)核心要素，曠視Brain++在總體架構(gòu)上分為三部分，包括深度學(xué)習(xí)框架MegEngine、深度學(xué)習(xí)云計(jì)算平臺(tái)MegCompute、以及數(shù)據(jù)管理平臺(tái)MegData。而天元?jiǎng)t是曠視Brain++最核心的組件。

據(jù)介紹，相比主流的開(kāi)源深度學(xué)習(xí)框架，天元降低了深度學(xué)習(xí)的門(mén)檻，可提升算法訓(xùn)練和部署的效率。

天元框架通過(guò)訓(xùn)練、推理一體化的機(jī)制省去模型轉(zhuǎn)換的過(guò)程，可直接將訓(xùn)練后的模型進(jìn)行推理，并保證跨設(shè)備的模型精度對(duì)齊，同時(shí)天元內(nèi)置自動(dòng)模型優(yōu)化、簡(jiǎn)化流程，減少因手工操作的機(jī)會(huì)，降低出錯(cuò)概率。

天元配置了Python 和 C++ 接口，支持動(dòng)態(tài)圖、靜態(tài)圖一鍵轉(zhuǎn)換和混合編程，可使用高級(jí)編程語(yǔ)言進(jìn)行圖優(yōu)化和圖編譯。而為了解決模型復(fù)現(xiàn)困難的問(wèn)題，天元?jiǎng)t支持 PyTorch Module 導(dǎo)入，可針對(duì)計(jì)算機(jī)視覺(jué)任務(wù)進(jìn)行優(yōu)化。

目前，曠視已在人工智能開(kāi)源開(kāi)放平臺(tái)OpenI啟智社區(qū)和全球最大的開(kāi)源社區(qū)GitHub上同步發(fā)布了天元 Alpha版源代碼，開(kāi)發(fā)者也可以通過(guò)天元官方網(wǎng)站的在線(xiàn)深度學(xué)習(xí)工具免費(fèi)調(diào)用算力、獲取最新數(shù)據(jù)集和訓(xùn)練腳本，進(jìn)行簡(jiǎn)單訓(xùn)練和試用。曠視為天元的開(kāi)發(fā)者還準(zhǔn)備了全球頂尖算法預(yù)訓(xùn)練模型的模型中心ModelHub，支持開(kāi)發(fā)者開(kāi)箱即用。

對(duì)于天元框架開(kāi)源后的發(fā)展規(guī)劃，曠視透露將在技術(shù)貢獻(xiàn)者的幫助下，于6月份推出 Beta 版本。

“也許下一代天元并不是由曠視的研發(fā)團(tuán)隊(duì)做出來(lái)的，而是與你一起共創(chuàng)出來(lái)。我們希望能夠吸引到一些優(yōu)秀的技術(shù)人員來(lái)給我們更好的建議，一起貢獻(xiàn)代碼、共建更好的深度學(xué)習(xí)框架。”曠視聯(lián)合創(chuàng)始人兼CTO唐文斌表示。