【TechWeb】3月25日,国内AI独角兽企业旷视科技今日正式宣布开源其深度学习框架“天元”。“天元”成为继百度飞桨之后,国内第二个开源的深度学习框架。
深度学习框架MegEngine是Brain++核心组件之一,旷视将其中文名称命名为“天元”。
Brain++由旷视科技完全自主研发,自2014年起一直应用于旷视企业内部,历经6年研发、迭代8个版本。
天元由旷视3名实习生在2013年启动研发,并于2014年正式完成上线。这款完全自研的深度学习框架历经6年,服务于旷视研究院1400余名AI开发者。
此次,旷视希望通过天元开源、开放的方式,降低开发者的门槛,逐步将AI生产力平台Brain++的能力传递给业界广大AI开发者。
旷视联合创始人、CTO唐文斌介绍AI的基础设施只有两部分组成——芯片平台和生产力平台。其中,芯片平台承载AI的计算,而AI算法的生产是一个系统工程,需要能够协同优化数据、算法、算力的平台级产品。
Brain++平台的意义是为了能像Visual Studio一样为用户提供更完善的集成开发环境,满足AI开发者从AI生产(输出算法模型)到应用(实现算法工程化封装)各环节中,一站式、全流程算法研发的切实需求,将产品从实验室原型到工业部署原本数周或数月的时间成本,缩短到小时级。
而作为Brain++核心组件,天元是一套训练推理一体化、动静态合一的工业级深度学习框架,架构上具体分为计算接口、图表示、优化与编译、运行时管理和计算内核五层。
天元开源项目的产品负责人田忠博详细介绍了天元的技术细节。天元配置了Python和C++ 接口,支持动态图、静态图一键转换和混合编程,可使用高级编程语言进行图优化和图编译。而为了解决模型复现困难的问题,天元则支持PyTorch Module导入,可针对计算机视觉任务进行优化。
田忠博介绍,目前,旷视已在中国新一代人工智能开源开放平台OpenI启智社区和全球最大的开源社区 GitHub上同步发布了天元Alpha版源代码,开发者也可以通过天元官方网站的在线深度学习工具免费调用算力、获取最新数据集和训练脚本,进行简单训练和试用。旷视为天元的开发者还准备了全球顶尖算法预训练模型的模型中心ModelHub,支持开发者开箱即用。
目前,天元Alpha版本已经在GitHub和OpenI平台中开源,6月份推出Beta版本。
唐文斌在开源介绍开源计划时也强调,“代码自己用和拿出来与人分享,会有巨大差距需要渐进式的升级,这次我们诚实地向大家宣布,当前开源的天元是Alpha版本,在向Beta版本和正式版本演进期间,我们希望更多优秀的开发者,以及院校以及企业能够参与进来,一起贡献code,提出建议,共建更好的天元。”