旷视开源深度学习框架天元 降低AI开发门槛

25日，旷视科技宣布正式开源新一代AI生产力平台Brain＋＋的核心深度学习框架旷视天元（MegEngine）。

图片来源：旷视科技

旷视联合创始人兼CTO唐文斌表示，“Brain＋＋ 平台的意义是为了能像 Visual Studio 一样为用户提供更完善的集成开发环境，满足 AI 开发者从 AI 生产（输出算法模型）到应用（实现算法工程化封装）各环节中，一站式、全流程算法研发的切实需求，将产品从实验室原型到工业部署原本数周或数月的时间成本，缩短到小时级。”

据介绍，旷视Brain＋＋总体架构包括深度学习框架MegEngine、深度学习云计算平台MegCompute、以及数据管理平台MegData。

图片来源：旷视科技

其中，天元是其最核心部分。它是一套训练推理一体化、动静态合一的工业级深度学习框架，架构上具体分为计算接口、图表示、优化与编译、运行时管理和计算内核五层。天元配置了Python 和 C＋＋ 接口，支持动态图、静态图一键转换和混合编程，并支持 PyTorch Module 导入。

天元开源项目的产品负责人田忠博表介绍，“天元对于 AI 开发者来说，就是一套从菜谱到美味的端到端一体化智能锅，覆盖从训练到推理再到部署装盘全流程。不仅自带菜谱（模型库）还支持自定义菜谱（模型导入），具备显示屏灵活可控、方便操作，特别适合既要求美味，又追求效率和体验的 AI 生产者。”

天元的核心目的是降低开发者门槛，以不断深化商业落地应用。唐文斌认强调，“代码自己用和拿出来与人分享，会有巨大差距需要渐进式的升级，这次我们诚实地向大家宣布，当前开源的天元是 Alpha 版本，在向 Beta 版本和正式版本演进期间，我们希望更多优秀的开发者，以及院校以及企业能够参与进来，一起贡献 code，提出建议，共建更好的天元。”