7月31日,新一代深度学习框架OneFlow正式迎来开源一周年。一年来,OneFlow已通过训练InsightFace、HugeCTR、GPT等模型,展示了其在各个方面的基准测试中比传统框架更高的效率。同时,OneFlow新增了多项功能,使得开发者的易用性体验大幅提升,包括:增加2-D SBP、流水并行,Checkpointing的新接口以及大量对齐PyTorch的接口,支持CUDA 11.2等。
作为自主研发的创新型产品,OneFlow也获得了多项行业认可,例如成为“AI中国”机器之心人工智能年度“最强开源技术生态TOP10”、就如全球知名创投研究机构CBInsights发布的“AI100”榜单,以及入选“2021世界人工智能大会SAIL奖TOP 30”。2021年2月4日,OneFlow完成由高瓴创投独家领投的5000万元人民币A轮融资。
当然,在深度学习框架领域,OneFlow还面临着激烈的竞争。其创始人袁进辉博士表示,改变深度学习框架的竞争格局绝非易事,不过目前深度学习框架领域的竞争是一场丛林探险,对未来有准确的判断对取胜至关重要,这也是小团队有机会超越大团队的原因。而OneFlow一直瞄准着超大规模模型,自动并行、去中心化等技术已经越来越多的被业界接认可和采用,相信未来的深度学习框架市场,OneFlow会有自己的一席之地。
据了解,接下来,OneFlow即将发布的版本会在单卡使用体验上和PyTorch对齐,进一步降低开发者上手门槛,同时,在多卡场景下推出的全新设计的编程接口,将给用户带来革命性开发体验。此外,通过补全推理和服务工具链、增加模型库、适配底层芯片等工作,推动OneFlow框架的生态更上一层楼。