日前,F(xiàn)acebook AI 基礎(chǔ)平臺(tái)部(AI Infra and Platform)副總 Bill Jia 在 Facebook Research Blog 發(fā)文宣布,亞馬遜將加入 ONNX 開放生態(tài),MXNet 將對(duì) ONNX 提供支持。
他表示,ONNX 開放生態(tài)于2017年9月啟動(dòng),旨在加強(qiáng)深度學(xué)習(xí)工具間的互操作性。
ONNX 是一種用于表示深度學(xué)習(xí)模型的標(biāo)準(zhǔn),支持模型在框架之間轉(zhuǎn)移,這是邁向開放生態(tài)的第一步。在這個(gè)生態(tài)中,AI 開發(fā)人員可以輕松地在最先進(jìn)的工具之間進(jìn)行轉(zhuǎn)移,選擇最合適的工具組合。
之后,亞馬遜也將為 ONNX 這種創(chuàng)新開放的生態(tài)系統(tǒng)做出貢獻(xiàn),開發(fā)人員將能構(gòu)建更豐富的模型,而不必?fù)?dān)心跨框架的互操作問題。
現(xiàn)在,亞馬遜的 MXNet 已經(jīng)支持 ONNX。ONNX-MXNet 是一個(gè)開源 Python 包,旨在將 ONNX 深度學(xué)習(xí)模型導(dǎo)入到 MXNet 中。作為一個(gè)功能全面的深度學(xué)習(xí)框架,MXNet 提供跨多種語言(如Python、Scala、R語言等)的 API 。
在 MXNet 加入 ONNX 之后,開發(fā)人員將有機(jī)會(huì)使用 PyTorch、CNTK、Caffe2 等框架來構(gòu)建和訓(xùn)練模型,再將模型導(dǎo)入 MXNet 中,來進(jìn)行更靈活、更優(yōu)化的運(yùn)行。
在未來,ONNX 的合作伙伴和社群將會(huì)繼續(xù)開發(fā) ONNX 格式,促進(jìn)對(duì)更多框架的支持。他們將會(huì)增強(qiáng)這個(gè)生態(tài)系統(tǒng)的互操作性,擴(kuò)展 ONNX MXNet 的功能,ONNX 很快將會(huì)支持 MXNet 核心 API 。開發(fā)人員將有機(jī)會(huì)在所有框架中自由地切換。