Apache MXNet是一個開源深度學習軟體框架,用於訓練及部署深度神經網路。MXNet具有可延伸性,允許快速模型訓練,並支援靈活的編程模型和多種程式語言(包括C++PythonJavaJuliaMatlabJavaScriptGoRScalaPerlWolfram語言)。

Quick Facts 開發者, 目前版本 ...
Apache MXNet
開發者Apache軟體基金會
目前版本
  • 1.9.1(2022年5月10日)[1]
編輯維基數據鏈結
原始碼庫 編輯維基數據鏈結
程式語言
作業系統WindowsmacOSLinux
類型機器學習深度學習
許可協定Apache License 2.0
網站mxnet.apache.org
Close

MXNet庫可以擴充到多GPU[2]和多台機器,並可移植。MXNet由公共雲提供商亞馬遜雲端運算服務(AWS)[3]Microsoft Azure[4]支援。亞馬遜把MXNet選為AWS的首選深度學習框架[5][6]。目前,MXNet受到英特爾、Dato、百度微軟沃爾夫勒姆研究公司以及卡內基·梅隆大學麻省理工學院華盛頓大學香港科技大學等研究機構的支援[7]

特色

Apache MXNet是一個極簡、靈活、可延伸的深度學習框架,支援深度學習模型,包括卷積神經網路(CNN)和長短期記憶網路(LSTM)。

可延伸性

MXNet分布於動態雲基礎架構上,使用分散式參數伺服器(基於卡內基·梅隆大學百度Google[8]),並且可以使用多GPU或多CPU實現近乎線性的擴充。

靈活性

MXNet支援命令式和符號式編程,讓使用指令式程式設計的開發者可以更輕鬆地上手深度學習,還可以更容易地跟蹤、除錯、儲存斷點,修改學習率等超參數或執行早停。

多程式語言支援

MXNet支援C++用於最佳化後端,以獲得大部分可用的GPU或CPU,以及支援Python、R語言、Scala、Julia、Perl、MATLAB和JavaScript,用於為開發人員提供簡單的前端。

可移植性

MXNet支援將受過訓練的模型高效部署到低階裝置,例如行動裝置(使用Amalgamation[9])、物聯網裝置(使用AWS Greengrass)、無伺服器計算(使用AWS Lambda)或容器。這些低階環境只有效能較弱的CPU或有限的主記憶體(RAM),並且應能使用在更高端環境(如基於GPU的叢集)上訓練的模型。

參見

參考資料

Wikiwand in your browser!

Seamless Wikipedia browsing. On steroids.

Every time you click a link to Wikipedia, Wiktionary or Wikiquote in your browser's search results, it will show the modern Wikiwand interface.

Wikiwand extension is a five stars, simple, with minimum permission required to keep your browsing private, safe and transparent.