1. 首页
  2. 人工智能

旷视开源“天元”MegEngine,共建深度学习框架

速途网3月25日讯(报道:乔志斌)今日,旷视召开Brain++ • 天元深度学习框架开源发布会,旷视联合创始人兼CTO唐文斌正式宣布将“天元”(MegEngine)的代码开源。

旷视开源“天元”MegEngine,共建深度学习框架

据介绍,“天元”始于2014年,是一个旷视自主研发的工业级深度学习框架,前身是旷视内部自用的算法训练推理引擎MegEngine,由旷视3名实习生在2013年启动研发,并与2014年正式完成上线。目前,天元开源的代码有35万行,C++代码占大部分,其次是CUDA、Python;它采用Apache License2.0,方便大家使用,包括商业的场景。经过六年的工业验证之后, 天元针对这次开源做了特别的大型升级,以全新的面目在新一代人工智能开放平台OpenI和Gihtub上与大家见面。

旷视开源“天元”MegEngine,共建深度学习框架

​ 天元框架通过训练、推理一体化的机制省去模型转换的过程,可直接将训练后的模型进行推理,并保证跨设备的模型精度对齐,同时天元内置自动模型优化、简化流程,减少因手工操作的机会,降低出错概率。除此之外,天元还配置了 Python 和 C++ 接口,支持动态图、静态图一键转换和混合编程,可使用高级编程语言进行图优化和图编译。在运行时管理环节,天元具备执行流与调度器,采用动态、静态内存分配并存的方式,并通过独创的自动的亚线性内存管理优化器可以得到更好的内存优化效果;在底层设计上,天元的计算内核不仅兼容主流计算设备,且支持多机多卡和分布式训练。而为了解决模型复现困难的问题,天元则支持 PyTorch Module 导入,可针对计算机视觉任务进行优化。

相比主流的开源深度学习框架,天元具有训练推理一体化、动静合一、兼容并包、灵活高效的显性优势,无论对于深度学习”老司机“,还是刚刚转型AI开发者的用户来说都非常友好,极大程度上降低了深度学习的门槛,提升了算法训练和部署的效率。

对于天元框架开源后的发展规划,旷视透露将在技术贡献者的帮助下,于6月份推出 Beta 版本。“也许下一代天元并不是由旷视的研发团队做出来的,而是与你一起共创出来。我们希望能够吸引到一些优秀的技术人员来给我们更好的建议,一起贡献代码、共建更好的深度学习框架。”唐文斌表示。