中国AI开源的新生力量!旷视“天元”深度学习框架全解读( 三 )



在此背景下 , 旷视选择开源AI框架 , 不仅是对其解决算法无限性这一长期战略的探索 , 同时旷视作为目前国内开源企业中唯一以AI为主业的公司 , 也期望能够通过开源让更多的合作伙伴实现AI+转型 , 进一步推动AI技术的加速落地 。
中国AI开源的新生力量!旷视“天元”深度学习框架全解读
文章图片

中国AI开源的新生力量!旷视“天元”深度学习框架全解读
文章图片

六年打磨 , 天元深度学习框架的前世今生

其实早在2014年 , 旷视就开始着手“天元”深度学习框架的研发 , 当时它的名字叫做MegEngine 。
为了让MegEngine早日跟上公司业务的发展进度 , 研究人员不断加速优化和研发 , 成功在2015年年中实现了MegEngine与公司内部所有业务的接轨 。 当时 , 旷视所有业务线上的模型 , 都已全部更换成MegEngine训练出来的版本 。
但MegEngine的发展并非一帆风顺 。
2015年11月 , 谷歌TensorFlow的正式开源 , 对当时已逐渐走入稳定发展期的MegEngine带来了巨大冲击 , 几乎打乱了旷视自研框架的开发节奏 。 面对国外科技巨头的压力 , 自研框架的路子是坚持还是放弃 , 成为当时旷视内部一场激烈持久的讨论 。
终于 , 在经过一次大规模的性能评测后 , 旷视的研究人员发现 , 当时刚开源的TensorFlow性能并不理想 , 比旷视的自研框架还要慢上10倍 。
正是这一发现 , 让旷视更加坚定不移地踏上自研框架的道路 , 并一路狂飙突进 , 不断克服研发道路上的阻碍 , 相继研发了深度学习云计算平台MegCompute和数据管理平台MegData , 与深度学习框架共同构成了AI生产力平台Brain++ 。
因此 , Brain++是一个集算法、算力和数据为一体的AI底层平台 , 拥有数据处理、清洗、管理 , 以及大规模分布式训练和计算等能力 , 帮助旷视开发了大量部署于云端、移动端、边缘端等全计算平台的先进深度神经网络 。
值得一提的是 , 在Brain++的数据、算法和算力三大核心架构中 , 最为核心开源了的深度学习框架天元MegEngine 。
中国AI开源的新生力量!旷视“天元”深度学习框架全解读
文章图片

随着AI产业发展的浪潮越来越汹涌 , 为了进一步加速AI技术落地 , 给更多开发者和企业用户提供开发选择 , 旷视从2019年起决定开源Brain++最核心的深度学习框架MegEngine , 并为它了一个全新的名字——天元 。

“天元”意味着围棋中棋盘的正中心 , 也代表着古代数学概念中所有复杂方程式的基础 , 还是古代天文学中指代万物本源和开始的北极星 。
“天元”这一名字的寓意背后 , 既表达了旷视希望MegEngine成为AI行业发展壮大的基石和重要组成部分 , 也透露出它之于旷视技术战略布局的最核心地位 , 是其AI宏图中最为关键和重要的底层基础设施 。
在准备开源的一年时间里 , 旷视对MegEngine的整个框架进行了脱胎换骨般的基础性升级和改动 , 研究人员对框架的所有代码进行了梳理和重组 , 并引入了动静合一等新特性 , 同时还进一步打磨对外接口 , 使其更符合广大开发者们的使用习惯 。
直至昨日 , 前后历时6年产业实践和工业验证的天元MegEngine , 终于开源面世 。


推荐阅读