旷视开源自研深度学习框架“天元” 助力行业打造AI基础设施

  • 日期:04-02
  • 点击:(585)


3月25日下午,师旷科技宣布将正式开放师旷田原公司的核心深度学习框架,这也使师旷科技成为世界上第一家开放底层框架的人工智能企业。

据报道,天元的前身是MegEngine,是师旷内部使用的算法训练和推理引擎。三名来自师旷的实习生于2013年开始研发,并于2014年在网上正式完成。作为师旷科技的基础技术框架,该框架在过去6年中支持了公司各种产品和业务的落地,也为师旷研究院1400多名人工智能开发者提供了服务。

师旷科技联合创始人兼首席技术官唐文斌说,师旷是世界上第一个用深度学习方法进行人工智能商业探索的科技企业。然而,由于缺乏早期的人工智能开发工具,师旷依靠自主研发,解决了从0到1的算法生成问题。

有了这个经验,唐文斌非常清楚,真正影响行业速度的不是算法、软件和硬件,也不是解决方案,而是人工智能基础设施。他说,“人工智能的基础设施由两部分组成:芯片平台和生产力平台。其中,芯片平台承载人工智能的计算。人工智能算法的产生是一项系统工程,需要能够协调优化数据、算法和计算力的平台级产品。”

唐文斌,以前师旷科技在天元的基础上建立了自己的基础设施Brain,现在也希望向更多的企业和开发商开放,以促进整个人工智能行业的繁荣。

天元开源项目负责人、研究院高级技术总监田表示,在传统的深度学习研发中,训练框架和推理框架这两个框架往往需要从原型到生产部署分别设计和调用,导致模型训练和推理转换过程中无法解释的性能或精度损失,需要开发人员手动优化,计算平台部署过程中出现的各种问题无法追溯。

天元框架在设计之初就避免了这样的问题。通过训练和推理相结合的机制消除模型转换的过程,可以直接对训练好的模型进行推理,保证模型在设备间的精确对准。同时,天元内置的自动模型优化和简化过程可以减少人工操作的机会和出错概率。

此外,天元配有Python和C接口,支持动态图形、静态图形一键转换和混合编程,可以使用高级编程语言进行图形优化和图形编译。

田表示,目前已经在开放源码平台人工智能的开源社区和世界上最大的开源社区?天元阿尔法的源代码同时在GitHub上发布。开发者还可以使用天元官方网站的在线深度学习工具调用计算力量,获取最新的数据集和培训脚本,进行简单的培训和试用。

师旷透露,在技术人员的帮助下,天元框架的测试版将于6月发布。

av天天看