中国AI开源的新生力量！旷视“天元”深度学习框架全解读( 四 ) 旷视宣布正式开源

同时，旷视云服务业务资深副总裁赵立威在发布会上表示，旷视将面向高校师生、企业开发者逐步开放Brain++的核心能力，为企业用户提供AI生产全流程服务。

实际上，旷视的开源定位与大多数开源玩家的定位不太一样，最显著的差异在于旷视是目前所有AI开源公司中，唯一一家以AI为主业的公司。
“天元是为了工业级、研究院规模的研发机构而开发的算法研发基础组件。 ”旷视研究院高级技术总监田忠博谈到，天元所具备的训练推理一体化等核心特点，都是为了解决开发者现实研发过程中的痛点而存在。
至此，旷视天元MegEngine的AI开源宏图正朝着未来缓缓铺开。

文章图片

文章图片

五大架构+四大技术优势助力多产业AI化发展

历时6六年自主研发和打磨的天元深度学习框架到底是什么？
从架构上看，天元具体分为计算接口、图表示、优化与编译、运行时管理和计算内核五层。

文章图片

若说谷歌TensorFlow采用利于部署的静态图更适用于工业界，而Facebook PyTorch采用灵活且方便调试的动态图更适合学术科研。那么旷视的天元则在兼具了双方特性的过程中，找到了一个的平衡点。

“天元是一个训练和推理在同一个框架、同一个体系内完整支持的设计。 ”田忠博谈到，天元从实际问题出发，提供了许多深度学习和AI落地问题的解决方案。
基于这些创新性的框架设计，天元深度学习框架拥有推理训练一体化、动静合一、兼容并包和灵活高效四大优势。
【中国AI开源的新生力量！旷视“天元”深度学习框架全解读】1、训练推理一体化
天元既能够支持开发者进行算法训练，同时其训练得到的模型，还可以直接用于产品的推理和封装，无需进行多余的模型转换。这极大地简化了算法开发流程，实现速度和精度的无损迁移。
与此同时，天元在模型部署时还能够自动优化模型，自动帮助开发者删除冗余代码。
2、动静合一
天元将动态图的简单灵活，与静态图的高性能优势进行整合，能在充分利用动态图模型训练优势的同时，通过动静态一键转换功能，以静态图的形式完成生产和部署。
此外，天元还支持动静态的混合编程，进一步提高其灵活性。
3、兼容并包
天元部署了Pythonic的API和PyTorch Module功能，支持模型直接导入，进一步降低框架迁移的入门门槛和学习成本。
同时，它内置高性能计算机视觉算子和算法，能够深度优化计算机视觉相关模型训练和应用。
4、灵活高效
在部署方面，天元拥有多平台多设备适应能力，其内置算子能够在推理或生产环境中充分利用多核优势，灵活调用设备算力，十分适用于大模型算法训练。