【资料图】
3月25日下午,迪法恩科技召开线上发布会,宣布新一代AI生产力平台Brain的核心深度学习框架MegEngine。
据了解,天元的前身是MegEngine,一个内部使用的算法训练和推理引擎。三位实习生2013年开始研发,2014年正式上线。这种自研的深度学习框架,支撑了蔑视六年在国际AI大赛上的表现,也支撑了公司产品和业务的落地。目前,它为蔑视研究所的1400多名人工智能开发人员提供服务。
发布会上,Defiance联合创始人兼CTO唐文斌正式宣布MegEngine代码开源,并指出它是集训练与推理、动态与静态于一体的工业级深度学习框架,中文名为天元。
据介绍,在传统的深度学习研发中,从原型到生产部署,往往需要将训练框架和推理框架分开设计和调用,导致模型在训练和推理转换过程中出现无法解释的性能或精度损失,需要开发人员手动优化,计算平台部署中算法的各种问题无法追溯。
天元框架避免了这个问题,通过训练和推理一体化的机制省略了模型转换的过程,可以直接对训练好的模型进行推理,保证模型跨设备的精确对齐。同时,天元内置的自动模型优化简化了流程,减少了人工操作的机会,降低了出错的概率。
此外,天元还配备了Python和C接口,支持动静图一键转换和混合编程,可以使用高级编程语言优化编译图。在运行时管理上,天元有执行流和调度器,采用动态和静态内存分配并存的方式,通过自动次线性内存管理优化器可以获得更好的内存优化效果。在底层设计上,天元的计算内核兼容主流计算设备,支持多机多卡、分布式训练。为了解决模型复制难的问题,天元支持导入PyTorch Module,可以针对计算机视觉任务进行优化。
目前,Defiance已经在中国新一代人工智能开源平台OpenI奇智社区和开源社区GitHub上同步发布了天元Alpha源代码。开发者也可以使用天元官网的在线深度学习工具调用算力,获取最新的数据集和训练脚本进行简单的训练和试用。
同时,被视为天元的开发者已经准备了算法预训练模型ModelHub,支持开发者开箱即用。据悉,对于天元框架开源后的发展计划,轻蔑透露,在技术贡献者的帮助下,Beta版将于6月推出。(定西)
郑重声明:本文版权归原作者所有,转载文章仅为传播更多信息之目的,如作者信息标记有误,请第一时候联系我们修改或删除,多谢。
标签: 上一篇:预计2020/21年度Airpods出货量将达到1.03亿/1.41亿台下一篇:最后一页关键词:
质检
推荐