mxnet

MXNet 是亞馬遜（Amazon）選擇的深度學習庫，並且也許是最優秀的庫之一。

基本介紹

• 中文名：MXNet
• 外文名：MXNet
• 版本：1.0.0

概述

MXNet 是亞馬遜（Amazon）選擇的深度學習庫。它擁有類似於 Theano 和 TensorFlow 的數據流圖，為多 GPU 配置提供了良好的配置，有著類似於 Lasagne 和 Blocks 更高級別的模型構建塊，並且可以在你可以想像的任何硬體上運行（包括手機）。對 Python 的支持只是其冰山一角—MXNet 同樣提供了對 R、Julia、C++、Scala、Matlab，和 Javascript 的接口。

1.0新特性

增強了sparse.dot運算符的性能。

MXNet現在自動設定OpenMP，以便在未設定NUM_OMP_THREADS時使用所有可用的CPU核心以最大限度地提高CPU利用率。

一元運算符和二元運算符現在可以避免在小型陣列上使用OpenMP，因為使用OpenMP實際上會損害多執行緒開銷。

在CPU上顯著提高了broadcast_add，broadcast_mul等的性能。

使用NCCL 2.1版或更新版本時，建議將環境變數NCCL_LAUNCH_MODE設定為PARALLEL。

MXNet可以加速任何數值計算[1]。神經網路中，每一層由一個線性函式和一個非線性變換組成。 在MXNet中，我們可以將其表示為

hidden_linear = mx.sym.dot(X, W)hidden_activation = mx.sym.tanh(hidden_linear)