阿裡雲機器學習平台 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架構的訓練任務。