近日,一流科技OneFlow深度学习框架正式和寒武纪思元270智能芯片及加速卡达成适配。这一合作达成,标志着OneFlow新一代深度学习框架的生态进一步得到完善,与新兴AI芯片进一步融合。
目前,OneFlow 通过集成寒武纪的高性能机器学习库,已经支持主流神经网络算子,并能支持在各类应用场景下的模型部署,如分类、检测、分割、风格转化等模型。下一步,双方将持续深化合作,以“大规模分布式”为目标,积极探索OneFlow深度学习框架和寒武纪其他产品的打通,丰富训练模型和应用场景。
一流科技的OneFlow深度学习框架被称为“新一代深度学习框架”,其核心优势在于用几行代码就可以完成其他主流框架需要深度定制才能做到的“超大规模分布式模型训练”。
OneFlow秉承“效率为王”的开发理念,拥有完全自主知识产权,所有代码全部自研。其独创的自动数据模型混合并行、静态调度、去中心化和全链路异步流式执行四大核心技术,使用纯软件技术完美地解决了大数据、大模型、大计算所带来的异构集群分布式横向扩展挑战,技术水平世界领先。在2020年5月由中国信通院发布的《首轮开源深度学习软件框架测试报告》中,在同样的算法和硬件条件下,OneFlow深度学习框架性能指标大幅领先国外主流框架。
寒武纪思元270集成了寒武纪在芯片架构领域的一系列创新性技术,基于16nm先进工艺制造,处理非稀疏深度学习模型的理论峰值性能提升至上一代MLU100的4倍,达到128TOPS(INT8),同时兼容INT4和INT16运算,理论峰值分别达到256TOPS和64TOPS,并支持浮点运算和混合精度运算。