3月25日下午消息,曠視通過線上發(fā)布會正式宣布開源新一代AI生產力平臺Brain++的核心深度學習框架曠視天元(MegEngine)。
天元的前身是曠視內部自用的算法訓練推理引擎MegEngine,由曠視3名實習生在2013年啟動研發(fā),并與2014年正式完成上線,支撐了曠視6年以來在國際AI競賽的表現(xiàn),以及公司產品和業(yè)務的落地,目前服務于曠視研究院1400余名AI開發(fā)者。
Brain++是曠視自主研發(fā)的端到端人工智能算法平臺,具備大規(guī)模算法研發(fā)能力。針對框架、算力和數(shù)據(jù)三個核心要素,曠視Brain++在總體架構上分為三部分,包括深度學習框架MegEngine、深度學習云計算平臺MegCompute、以及數(shù)據(jù)管理平臺MegData。而天元則是曠視Brain++最核心的組件。
據(jù)介紹,相比主流的開源深度學習框架,天元降低了深度學習的門檻,可提升算法訓練和部署的效率。
天元框架通過訓練、推理一體化的機制省去模型轉換的過程,可直接將訓練后的模型進行推理,并保證跨設備的模型精度對齊,同時天元內置自動模型優(yōu)化、簡化流程,減少因手工操作的機會,降低出錯概率。
天元配置了Python 和 C++ 接口,支持動態(tài)圖、靜態(tài)圖一鍵轉換和混合編程,可使用高級編程語言進行圖優(yōu)化和圖編譯。而為了解決模型復現(xiàn)困難的問題,天元則支持 PyTorch Module 導入,可針對計算機視覺任務進行優(yōu)化。
目前,曠視已在人工智能開源開放平臺OpenI啟智社區(qū)和全球最大的開源社區(qū)GitHub上同步發(fā)布了天元 Alpha版源代碼,開發(fā)者也可以通過天元官方網站的在線深度學習工具免費調用算力、獲取最新數(shù)據(jù)集和訓練腳本,進行簡單訓練和試用。曠視為天元的開發(fā)者還準備了全球頂尖算法預訓練模型的模型中心ModelHub,支持開發(fā)者開箱即用。
對于天元框架開源后的發(fā)展規(guī)劃,曠視透露將在技術貢獻者的幫助下,于6月份推出 Beta 版本。
“也許下一代天元并不是由曠視的研發(fā)團隊做出來的,而是與你一起共創(chuàng)出來。我們希望能夠吸引到一些優(yōu)秀的技術人員來給我們更好的建議,一起貢獻代碼、共建更好的深度學習框架?!睍缫暵?lián)合創(chuàng)始人兼CTO唐文斌表示。