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

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

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