旷视正式宣布开源深度学习框架“天元” 降低AI开发门槛

小熊在线 马甲 | 2020年03月26日
【小熊在线网 新闻报道】3月25日,旷视科技宣布开源新一代AI生产力平台Brain++的核心深度学习框架旷视天元(MegEngine)。天元框架通过训练、推理一体 ......

【小熊在线网 新闻报道】3月25日,旷视科技宣布开源新一代AI生产力平台Brain++的核心深度学习框架旷视天元(MegEngine)。天元框架通过训练、推理一体化的机制省去模型转换的过程,可直接将训练后的模型进行推理,并保证跨设备的模型精度对齐,同时天元内置自动模型优化、简化流程,减少因手工操作的机会,降低出错概率。此外,天元配置了Python和C++接口,支持动态图、静态图一键转换和混合编程,可使用高级编程语言进行图优化和图编译。在运行时管理环节,天元具备执行流与调度器,采用动态、静态内存分配并存的方式,并通过自动的亚线性内存管理优化器可以得到更好的内存优化效果;在底层设计上,天元的计算内核兼容主流计算设备,且支持多机多卡和分布式训练。而为了解决模型复现困难的问题,天元则支持 PyTorch Module 导入,可针对计算机视觉任务进行优化。

标签:互联网

用户名:  密码:  没有注册?
网友评论:(请各位网友遵纪守法并注意语言文明,评论仅供参考不代表本站立场)