Categories
創造力

曠視開源深度學習框架天元 降低AI開發門檻


DoNews3月25日消息(記者 翟繼茹)25日,曠視科技宣布正式開源新一代AI生產力平台Brain++的核心深度學習框架曠視天元(MegEngine)。

image.png

曠視聯合創始人兼CTO唐文斌表示,“Brain++ 平台的意義是為了能像Visual Studio 一樣為用戶提供更完善的集成開發環境,滿足AI 開發者從AI 生產(輸出算法模型)到應用(實現算法工程化封裝)各環節中,一站式、全流程算法研發的切實需求,將產品從實驗室原型到工業部署原本數週或數月的時間成本,縮短到小時級。”

據介紹,曠視Brain++總體架構包括深度學習框架MegEngine、深度學習雲計算平台MegCompute、以及數據管理平台MegData。

image.png

其中,天元是其最核心部分。它是一套訓練推理一體化、動靜態合一的工業級深度學習框架,架構上具體分為計算接口、圖表示、優化與編譯、運行時管理和計算內核五層。天元配置了Python 和 C++ 接口,支持動態圖、靜態圖一鍵轉換和混合編程,並支持 PyTorch Module 導入。

天元開源項目的產品負責人田忠博表介紹,“天元對於AI 開發者來說,就是一套從菜譜到美味的端到端一體化智能鍋,覆蓋從訓練到推理再到部署裝盤全流程。不僅自帶菜譜(模型庫)還支持自定義菜譜(模型導入),具備顯示屏靈活可控、方便操作,特別適合既要求美味,又追求效率和體驗的AI 生產者。”

天元的核心目的是降低開發者門檻,以不斷深化商業落地應用。唐文斌認強調,“代碼自己用和拿出來與人分享,會有巨大差距需要漸進式的升級,這次我們誠實地向大家宣布,當前開源的天元是Alpha 版本,在向Beta 版本和正式版本演進期間,我們希望更多優秀的開發者,以及院校以及企業能夠參與進來,一起貢獻code,提出建議,共建更好的天元。”(完)