耗费资源打造优化6年，迭代到8.0版本，旷视技术体系的关键支柱，现在正式对外开源。
今天（3月25日）下午，中关村人工智能企业旷视科技举办线上发布会，旷视联合创始人兼CTO唐文斌宣布正式开源其AI生产力平台Brain++的核心组件——天元(MegEngine)。本次发布的天元为Alpha 版本，基于Apache License2.0，向外界共开源约35万行代码，包括C++、CUDA和Python代码。
什么是天元，为何开源？
天元始于2014年，是一个旷视自主研发的工业级深度学习框架，帮助旷视拿下27项全球AI竞赛冠军和MS COCO三连冠。旷视1400名研发人员使用天元，开发公司的所有AI产品。
天元结构示意图
“天元可帮助开发者做到协同设计训得好，高效系统训得快，大规模数据或模型训得动。”旷视首席科学家、研究院长孙剑在发布会上表示。
发布会上，唐文斌宣布将通过开源、开放的形式逐步把Brain++的能力释放出来，和行业共享。
作者系网易新闻·网易号“各有态度”签约作者
对于为什么选择开源，唐文斌介绍：“我们希望能够通过天元这样一个深度学习框架，能够作为中国人工智能的一个基石，能够帮助中国人工智能做得更好。”
而聚焦到旷视本身，开源亦是其基于算法向产业方向升级，打造AI基础设施战略的延伸。
经过六年的工业验证之后， 天元针对这次开源做了特别的大型升级，以全新的面目在新一代人工智能开放平台OpenI和Gihtub上与大家见面。此次天元开源的代码有35万行，C++代码占大部分，其次是CUDA、Python。它采用Apache License2.0，方便大家使用，包括商业的场景。
旷视联合创始人兼 CTO 唐文斌
唐文斌介绍，此次开源的天元是Alpha版本，基于Apache License2.0，囊括了大多数应用场景。接下来的6月份，旷视将发布天元的Beta版本，增加对ARM系列CPU支持、更多加速设备支持、量化和低比特计算支持等功能。9月份，旷视会发布正式1.0版本，全面支持主流计算设备，升级动态计算能力，优化训练推理全流程使用体验等。
天元的四大优势
天元是一个复杂的系统，它的目的是为了解决旷视多年遇到的AI产品开发困境。
天元有着训练推理一体、动静合一、兼容并包、灵活高效四大优势，还支持顶级研究团队的最前沿学术开发。以天元深度学习框架为基础，可以打造新一代AI生产力平台，让开发者更简单、更高效地进行AI生产力的研发。
一、 训练推理一体
AI产品的研发，从研究到原型再到生产，是一个非常复杂的流程，这造成很多问题，比如精度、速度不及预期。
天元「训推一体」图示
面对这种的问题，天元推出训练推理一体的方案，即训练推理可以同时进行，它有四个特点：
1、无需模型转换，直接使用训练模型进行推理。
2、速度和精度在推理时保持一致。
3、模型训练结束之后，用于不同设备时，可保障跨设备模型精度对齐。
4、天元内置自动模型优化模块，减少手工模型遇错处理，简化整个流程。
二、动静合一
深度学习框架大致分为两类，一类是静态框架，易于部署，但是难调适；另一类是动态框架，调适更方便，使用更灵活，但资源占用多。这是一个典型的鱼与熊掌的问题。
天元「动静合一」代码示例
而天元既可在动态下正确运行，又可切换到静态形态。 达成一种动静合一的效果
三、兼容并包
深度学习发展至今，框架众多，接口各异，如果开发者要复现模型，需要学习多种框架知识。这是一个巨大的学习成本。
天元「兼容并包」图示
面对框架接口各异，模型复现困难，学习成本高的问题，天元发展成了一个兼容并包的体系。
四、 灵活高效
AI公司通常需要非常多的计算设备和场景，如果要提供极致的AI产品和服务，就要在每种设备上打造极致的性能，这是非常大的挑战。
天元灵活高效图示
天元具备很强的多平台多设备适应能力，能通过类似汇编和指令重排等技术，使得天元内置算子能够在推理或生产环境中充分利用多核优势，灵活调用设备的计算力，尤其适用于大模型算法训练。
专家观点
人工智能深度学习框架则是最关键的人工智能基础设施之一。目前，主流的开源深度学习框架都来谷歌、Facebook、亚马逊国外科技巨头企业。 中国人工智能产业过度依赖国外开源代码，在核心技术上容易被“卡脖子”。
此次旷视选择开源自用6年的深度学习框架，同时将Brain++的AI生产力开放给广大企业用户，本质是将深度学习和高效开发算法、开发能力赋能给传统产业、中小企业和高校师生，是降低算法研发门槛、加速产业数字化转型的务实之举，有望加速推动人工智能成为经济和社会的基础设施。
来源丨综合整理
编辑丨武圣俊
点亮，
本文网址：
