速途網(wǎng)3月25日訊(報道:喬志斌)今日,曠視召開Brain++???天元深度學(xué)習(xí)框架開源發(fā)布會,曠視聯(lián)合創(chuàng)始人兼CTO唐文斌正式宣布將“天元”(MegEngine)的代碼開源。

據(jù)介紹,“天元”始于2014年,是一個曠視自主研發(fā)的工業(yè)級深度學(xué)習(xí)框架,前身是曠視內(nèi)部自用的算法訓(xùn)練推理引擎MegEngine,由曠視3名實習(xí)生在2013年啟動研發(fā),并與2014年正式完成上線。目前,天元開源的代碼有35萬行,C++代碼占大部分,其次是CUDA、Python;它采用Apache License2.0,方便大家使用,包括商業(yè)的場景。經(jīng)過六年的工業(yè)驗證之后, 天元針對這次開源做了特別的大型升級,以全新的面目在新一代人工智能開放平臺OpenI和Gihtub上與大家見面。

??天元框架通過訓(xùn)練、推理一體化的機制省去模型轉(zhuǎn)換的過程,可直接將訓(xùn)練后的模型進(jìn)行推理,并保證跨設(shè)備的模型精度對齊,同時天元內(nèi)置自動模型優(yōu)化、簡化流程,減少因手工操作的機會,降低出錯概率。除此之外,天元還配置了?Python?和?C++?接口,支持動態(tài)圖、靜態(tài)圖一鍵轉(zhuǎn)換和混合編程,可使用高級編程語言進(jìn)行圖優(yōu)化和圖編譯。在運行時管理環(huán)節(jié),天元具備執(zhí)行流與調(diào)度器,采用動態(tài)、靜態(tài)內(nèi)存分配并存的方式,并通過獨創(chuàng)的自動的亞線性內(nèi)存管理優(yōu)化器可以得到更好的內(nèi)存優(yōu)化效果;在底層設(shè)計上,天元的計算內(nèi)核不僅兼容主流計算設(shè)備,且支持多機多卡和分布式訓(xùn)練。而為了解決模型復(fù)現(xiàn)困難的問題,天元則支持?PyTorch Module?導(dǎo)入,可針對計算機視覺任務(wù)進(jìn)行優(yōu)化。
相比主流的開源深度學(xué)習(xí)框架,天元具有訓(xùn)練推理一體化、動靜合一、兼容并包、靈活高效的顯性優(yōu)勢,無論對于深度學(xué)習(xí)”老司機“,還是剛剛轉(zhuǎn)型AI開發(fā)者的用戶來說都非常友好,極大程度上降低了深度學(xué)習(xí)的門檻,提升了算法訓(xùn)練和部署的效率。
對于天元框架開源后的發(fā)展規(guī)劃,曠視透露將在技術(shù)貢獻(xiàn)者的幫助下,于6月份推出?Beta?版本。“也許下一代天元并不是由曠視的研發(fā)團(tuán)隊做出來的,而是與你一起共創(chuàng)出來。我們希望能夠吸引到一些優(yōu)秀的技術(shù)人員來給我們更好的建議,一起貢獻(xiàn)代碼、共建更好的深度學(xué)習(xí)框架?!碧莆谋蟊硎尽?/p>