MegEngine(天元) 深度学习框架开源项目

我要开发同款
匿名用户2020年03月30日
49阅读
开发技术C/C++
所属分类人工智能、机器学习/深度学习
授权协议Apache

作品详情

MegEngine(天元)是一个快速,可扩展,易于使用且支持自动求导的深度学习框架,中文名为“天元”,是旷视AI战略的重要组成部分,负责AI三要素(算法,算力,数据)中的“算法”。

特点:

一是“训练推理一体”。MegEngine支持多种硬件平台(CPU,GPU,ARM)。不同硬件上的推理框架和MegEngine的训练框架无缝衔接。部署时无需做额外的模型转换,速度/精度和训练保持一致,有效解决了AI落地中“部署环境和训练环境不同,部署难”的问题。

二是“动静合一”。动态图易调试,静态图好部署。鱼和熊掌如何兼得,是现代深度学习框架的核心诉求。MegEngine在静态图的基础上,逐渐加入支持完整动态图的功能。在动态模式下加速研发过程,无需改变模型代码一键切换至静态模式下的部署,为科研和算法工程师同时提供便利。

三是“兼容并包”。MegEngine的顶层API基于Python,采取了类似于PyTorch的风格。简单直接,易于上手,便于现有项目进行移植或整合。为更好地帮助学习实践,MegEngine同时提供了“开箱即用”的在线深度学习工具 MegStudio ,和汇聚了顶尖算法和模型的预训练模型集合 ModelHub 。

四是“灵活高效”。MegEngine底层的高性能算子库对于不同的硬件架构进行了深度适配和优化,并提供高效的亚线性内存优化策略,对于生产环境繁多的计算设备提供了极致的性能保证。高效易用的分布式训练实现能有效支持富有弹性的大规模训练。

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论