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

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唐文斌表示。