旷视开源「天元」，共创深度学习框架新纪元

 

3月25日，旷视召开Brain++ • 天元深度学习框架开源发布会，宣布「天元」开源。「天元」（MegEngine）始于2014年，是一个旷视自主研发的工业级深度学习框架，帮助旷视拿下27项全球AI竞赛冠军和MS COCO三连冠。旷视1400名研发人员使用它，开发公司的所有AI产品。

 

天元开源的代码有35万行，C++代码占大部分，其次是CUDA、Python；它采用Apache License2.0，方便大家使用，包括商业的场景。经过六年的工业验证之后， 天元针对这次开源做了特别的大型升级，以全新的面目在新一代人工智能开放平台OpenI和Gihtub上与大家见面。

 

天元的理念是「深度学习简单开发」，正好和旷视的基因「技术信仰价值务实」一一相映。天元之名取自围棋，以开源之名，向AlphaGO致敬，表明旷视秉持的技术信仰；同时天元寓意天地万物之元基，希望天元成为中国AI「新基建」的基石，彰显旷视坚守的价值务实。天元希望在技术与价值的双螺旋上升之中，与开发者一起，共创深度学习框架新纪元。

天元的整体架构

作为旷视全员使用的工业级深度学习框架，天元有着先进的整体架构，由上而下，它分为五个层次：1）「计算接口层」，2）「图表示层」，3）「优化与编译层」，4）「运行时管理层」，5）「计算内核层」。

 

天元架构示意图

最上面的「计算接口层」向外暴露了Python和C++接口，开发者借由这两种高级语言，使用天元编程，设计和研发原型系统，进行训练和推理。

 

第二层是「图表示」，它包含动态图和静态图的表示功能。第三层是「优化与编译」，它包括自动求导、图优化、图编译三个模块，是一个完整的、一体化的核心计算图，支撑着其上的计算接口、动态图、静态图的完整功能。

 

第四层是「运行时管理」，它包含计算调度和内存管理两个部分，前者把整个计算设备抽象为执行流，由调度器做合理调度࿱