阿里云機器學習平臺 PAI宣布集成國產深度學習框架 OneFlow
11月4日消息,在云棲大會上,阿里云機器學習平臺 PAI宣布集成自研深度學習框架OneFlow,進一步提升對國產算法框架的支持。
阿里云機器學習平臺PAI是一站式AI開發平臺,提供了豐富的機器學習組件和云原生開發工具,集成了國際主流的機器學習框架例如TensorFlow、PyTorch和Caffe等。同時,PAI具備開放性的技術架構,支持第三方社區框架集成到PAI平臺,以社區鏡像或自定義鏡像的方式提供給開發者。
將自研深度學習框架OneFlow集成到PAI上,也是對機器學習平臺PAI開放性的驗證,PAI可以在架構上實現包括對國際主流、國內自研在內的任何第三方深度學習框架的支持。
OneFlow 是一款易用、高效、高擴展,自主研發的深度學習框架。其算子和模塊編程接口與 PyTorch 兼容性極高,基于 PyTorch 搭建的常見模型幾乎不用修改即可在 OneFlow 上運行,遷移成本低。并且可以輕松切換動靜模式,模型調試階段享有和PyTorch 完全一樣的動態圖編程體驗;除此之外,OneFlow 計算性能卓越,在主流模型上最大有一個數量級的性能提升,極致系統級優化實現幾乎零運行期開銷。
目前,AI開發者已經可以在PAI的云原生開發環境中使用OneFlow進行機器學習訓練和離線推理。開發者在PAI的深度學習容器DLC中創建訓練任務時,只需要在“社區鏡像”列表里選擇OneFlow鏡像,任務提交后PAI就可以使用OneFlow框架執行深度學習訓練任務。開發者也可以在PAI的Data Science Workshop (DSW) 交互式開發工具中安裝OneFlow軟件并執行基于OneFlow框架的訓練任務。
除了通過合作集成的方式實現對第三方社區框架的支持,PAI平臺還允許開發者使用自定義鏡像來執行云原生訓練任務。開發者可以在深度學習容器DLC中選擇自己事先保存好的鏡像,或者指定自定義鏡像地址,在提交訓練任務時PAI就會使用該鏡像執行相應的任務。PAI平臺的開放性架構設計實現對任何第三方機器學習框架的支持,具有良好的可擴展性和豐富性,是基于云原生AI開發的強大平臺。
本站所有文章、數據、圖片均來自互聯網,一切版權均歸源網站或源作者所有。
如果侵犯了你的權益請來信告知我們刪除。郵箱:business@qudong.com