世界焦点！中国人工智能公司旷视科技开源深度学习框架‘MegEngine’

来源：互联网时间：2023-04-19 02:57:17

中国人工智能(MAI)公司旷视科技今天宣布，已开放其深度学习框架MegEngine，这是新一代AI生产力平台Brain的核心。

MegEngine是旷视科技自己使用的算法训练和推理引擎。它由三名旷视科技实习生于2013年创建，并于2014年正式启动。

(资料图片)

这一自行开发的深度学习框架支持了旷视科技在国际AI竞赛中的表现以及过去六年公司产品和业务的实施。目前服务于旷视科技学院1400多名AI开发者。

发布会上，旷视科技联合创始人兼CTO唐文斌正式宣布MegEngine的代码是开源的，并指出这是一个集训练与推理、动态与静态于一体的工业级深度学习框架。

在传统的深度学习研发中，从原型到生产部署，需要分别设计和调用训练框架和推理框架。

这会导致训练和模型推理的转换过程中出现无法解释的性能或精度损失，需要开发人员手动优化，无法追溯算法部署在计算平台上时出现的各种问题。

MegEngine框架避免了这样的问题。通过训练和推理的集成，可以省略模型转换的过程，直接推断出训练好的模型，并且可以跨设备调整模型的精度。

同时，MegEngine内置了自动模型优化功能，简化了流程，减少了人工操作的机会和出错的可能性。

此外，MegEngine还配备了Python和C接口，支持动态图、静态图一键转换和混合编程，可以使用高级编程语言进行图优化和图编译。

在运行时管理中，MegEngine有一个执行进程和一个调度程序。它采用动态和静态内存分配并存，通过自动次线性内存管理优化器可以得到更好的内存优化结果。

在基础设计上，MegEngine的计算内核兼容主流计算设备，支持多机多卡和分布式训练。

为了解决模型复制困难的问题，MegEngine支持导入PyTorch Module，可以针对计算机视觉任务进行优化。

目前，旷视科技已经在OpenI奇智社区和开源社区GitHub(中国新一代人工智能开源平台)发布了MegEngine Alpha的源代码。

开发者还可以使用MegEngine官网上的在线深度学习工具，自由调用计算能力，获取最新的数据集和训练脚本，进行简单的训练和试用。

同时，旷视科技为MegEngine开发者准备了算法预训练模型ModelHub，支持开发者开箱即用。

至于MegEngine框架开源后的开发计划，旷视科技透露将在技术贡献者的帮助下，于6月推出测试版。

关键词：