旷视开源深度学习框架天元 缩短AI算法训练周期

来源:新浪科技 | 2020-03-25 16:39:30 |

3月25日下午消息,旷视通过线上发布会正式宣布开源新一代AI生产力平台Brain++的核心深度学习框架旷视天元(MegEngine)。

天元的前身是旷视内部自用的算法训练推理引擎MegEngine,由旷视3名实习生在2013年启动研发,并与2014年正式完成上线,支撑了旷视6年以来在国际AI竞赛的表现,以及公司产品和业务的落地,目前服务于旷视研究院1400余名AI开发者。

Brain++是旷视自主研发的端到端人工智能算法平台,具备大规模算法研发能力。针对框架、算力和数据三个核心要素,旷视Brain++在总体架构上分为三部分,包括深度学习框架MegEngine、深度学习云计算平台MegCompute、以及数据管理平台MegData。而天元则是旷视Brain++最核心的组件。

据介绍,相比主流的开源深度学习框架,天元降低了深度学习的门槛,可提升算法训练和部署的效率。

天元框架通过训练、推理一体化的机制省去模型转换的过程,可直接将训练后的模型进行推理,并保证跨设备的模型精度对齐,同时天元内置自动模型优化、简化流程,减少因手工操作的机会,降低出错概率。

天元配置了Python 和 C++ 接口,支持动态图、静态图一键转换和混合编程,可使用高级编程语言进行图优化和图编译。而为了解决模型复现困难的问题,天元则支持 PyTorch Module 导入,可针对计算机视觉任务进行优化。

目前,旷视已在人工智能开源开放平台OpenI启智社区和全球最大的开源社区GitHub上同步发布了天元 Alpha版源代码,开发者也可以通过天元官方网站的在线深度学习工具免费调用算力、获取最新数据集和训练脚本,进行简单训练和试用。旷视为天元的开发者还准备了全球顶尖算法预训练模型的模型中心ModelHub,支持开发者开箱即用。

对于天元框架开源后的发展规划,旷视透露将在技术贡献者的帮助下,于6月份推出 Beta 版本。

“也许下一代天元并不是由旷视的研发团队做出来的,而是与你一起共创出来。我们希望能够吸引到一些优秀的技术人员来给我们更好的建议,一起贡献代码、共建更好的深度学习框架。”旷视联合创始人兼CTO唐文斌表示。

关键词: 旷视科技