新一代深度學習框架OneFlow迎來開源一周年,多項更新正在規劃中
來源:互聯網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框架的生態更上一層樓。