旷视开源深度学习框架天元 降低AI开发门槛
25日,旷视科技宣布正式开源新一代AI生产力平台Brain++的核心深度学习框架旷视天元(MegEngine)。
图片来源:旷视科技
旷视联合创始人兼CTO唐文斌表示,“Brain++ 平台的意义是为了能像 Visual Studio 一样为用户提供更完善的集成开发环境,满足 AI 开发者从 AI 生产(输出算法模型)到应用(实现算法工程化封装)各环节中,一站式、全流程算法研发的切实需求,将产品从实验室原型到工业部署原本数周或数月的时间成本,缩短到小时级。”
据介绍,旷视Brain++总体架构包括深度学习框架MegEngine、深度学习云计算平台MegCompute、以及数据管理平台MegData。
图片来源:旷视科技
其中,天元是其最核心部分。它是一套训练推理一体化、动静态合一的工业级深度学习框架,架构上具体分为计算接口、图表示、优化与编译、运行时管理和计算内核五层。天元配置了Python 和 C++ 接口,支持动态图、静态图一键转换和混合编程,并支持 PyTorch Module 导入。
天元开源项目的产品负责人田忠博表介绍,“天元对于 AI 开发者来说,就是一套从菜谱到美味的端到端一体化智能锅,覆盖从训练到推理再到部署装盘全流程。不仅自带菜谱(模型库)还支持自定义菜谱(模型导入),具备显示屏灵活可控、方便操作,特别适合既要求美味,又追求效率和体验的 AI 生产者。”
天元的核心目的是降低开发者门槛,以不断深化商业落地应用。唐文斌认强调,“代码自己用和拿出来与人分享,会有巨大差距需要渐进式的升级,这次我们诚实地向大家宣布,当前开源的天元是 Alpha 版本,在向 Beta 版本和正式版本演进期间,我们希望更多优秀的开发者,以及院校以及企业能够参与进来,一起贡献 code,提出建议,共建更好的天元。”