旷视天元 MegEngine

旷视天元开源深度学习框架 MegEngine 是旷视自主研发的国产工业级深度学习框架，是旷视新一代 AI 生产力平台 Brain++ 的最核心组件，在 2020 年 3 月正式向全球开发者开源。MegEngine 凭借其训练推理一体、超低硬件门槛和全平台高效推理 3 大核心优势，能够帮助企业与开发者大幅节省产品从实验室原型到工业部署的流程，真正实现小时级的转化能力。

MegEngine 核心特性

训练推理一体

• 一套内核支撑训练到推理，无需模型转换，精度损失最小化。
• 前后处理可放入计算图推理，训练推理精确对齐，python C++ 不用写两遍。
• traced module → megengine lite → megflow，从模型到高并发视频流处理服务只需几行python代码。

超低硬件门槛

• DTR 算法减少 75% 显存占用，1080 也能训 transformer。
• 自研 pushdown 内存分配算法，带来最低的内存/显存占用。
• 自动代码裁剪可使部署文件 binary size 下降 10 倍，有效降低推理硬件成本。

全平台高效推理

• 高效的推理性能，在各类 CPU、GPU 上均可享受到极致性能体验。
• 自动 layout、kernel 算法选择机制，轻松达到最优推理性能。
• 跨平台模型精度对齐，一套代码走天下。

联系我们

官网：https://www.megengine.org.cn/

论坛：https://discuss.megengine.org.cn/

GitHub：https://github.com/MegEngine

邮箱：[email protected]

QQ 交流群：群号1029741705