当我们谈论 MegEngine 时,我们在谈论什么
众所周知,开源深度学习框架旷视天元(MegEngine)是旷视自主研发的国产工业级深度学习框架 ,是旷视新一代 AI 生产力平台 Brain++ 的最核心组件,在2020年3月正式向全球开发者开源。MegEngine 凭借其训练推理一体、超低硬件门槛和全平台高效推理 3 大核心优势,帮助企业与开发者大幅节省产品从实验室原型到工业部署的流程,真正实现小时级的转化能力。
MegEngine
自上向下包含三个层次:Imperative
、MegBrain
和 MegDNN
。它们的角色定位分别是:
Imperative
:MegEngine
为动态训练设计的一套新接口,负责处理动态图运行时(Imperative Runtime
)。MegBrain
:负责处理静态图运行时(Graph Runtime
)。MegDNN
:MegEngine
的底层计算引擎。
下面再从下到上介绍一下这三个模块在 MegEngine
中的作用。
MegEngine 的底层计算引擎 —— MegDNN
MegDNN
在 MegEngine
中扮演的角色