11月4日消息，在云棲大會上，阿里云機器學習平臺 PAI宣布集成自研深度學習框架OneFlow，進一步提升對國產(chǎn)算法框架的支持。

阿里云機器學習平臺PAI是一站式AI開發(fā)平臺，提供了豐富的機器學習組件和云原生開發(fā)工具，集成了國際主流的機器學習框架例如TensorFlow、PyTorch和Caffe等。同時，PAI具備開放性的技術架構(gòu)，支持第三方社區(qū)框架集成到PAI平臺，以社區(qū)鏡像或自定義鏡像的方式提供給開發(fā)者。

將自研深度學習框架OneFlow集成到PAI上，也是對機器學習平臺PAI開放性的驗證，PAI可以在架構(gòu)上實現(xiàn)包括對國際主流、國內(nèi)自研在內(nèi)的任何第三方深度學習框架的支持。

OneFlow 是一款易用、高效、高擴展，自主研發(fā)的深度學習框架。其算子和模塊編程接口與 PyTorch 兼容性極高，基于 PyTorch 搭建的常見模型幾乎不用修改即可在 OneFlow 上運行，遷移成本低。并且可以輕松切換動靜模式，模型調(diào)試階段享有和PyTorch 完全一樣的動態(tài)圖編程體驗；除此之外，OneFlow 計算性能卓越，在主流模型上最大有一個數(shù)量級的性能提升，極致系統(tǒng)級優(yōu)化實現(xiàn)幾乎零運行期開銷。

目前，AI開發(fā)者已經(jīng)可以在PAI的云原生開發(fā)環(huán)境中使用OneFlow進行機器學習訓練和離線推理。開發(fā)者在PAI的深度學習容器DLC中創(chuàng)建訓練任務時，只需要在“社區(qū)鏡像”列表里選擇OneFlow鏡像，任務提交后PAI就可以使用OneFlow框架執(zhí)行深度學習訓練任務。開發(fā)者也可以在PAI的Data Science Workshop (DSW) 交互式開發(fā)工具中安裝OneFlow軟件并執(zhí)行基于OneFlow框架的訓練任務。

除了通過合作集成的方式實現(xiàn)對第三方社區(qū)框架的支持，PAI平臺還允許開發(fā)者使用自定義鏡像來執(zhí)行云原生訓練任務。開發(fā)者可以在深度學習容器DLC中選擇自己事先保存好的鏡像，或者指定自定義鏡像地址，在提交訓練任務時PAI就會使用該鏡像執(zhí)行相應的任務。PAI平臺的開放性架構(gòu)設計實現(xiàn)對任何第三方機器學習框架的支持，具有良好的可擴展性和豐富性，是基于云原生AI開發(fā)的強大平臺。