MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systems
U. Washington
arXiv:1512.01274 [cs.DC], (3 Dec 2015)
@article{chen2015mxnet,
title={MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systems},
author={Chen, Tianqi and Li, Mu and Li, Yutian and Lin, Min and Wang, Naiyan and Wang, Minjie and Xiao, Tianjun and Xu, Bing and Zhang, Chiyuan and Zhang, Zheng},
year={2015},
month={dec},
archivePrefix={"arXiv"},
primaryClass={cs.DC}
}
MXNet is a multi-language machine learning (ML) library to ease the development of ML algorithms, especially for deep neural networks. Embedded in the host language, it blends declarative symbolic expression with imperative tensor computation. It offers auto differentiation to derive gradients. MXNet is computation and memory efficient and runs on various heterogeneous systems, ranging from mobile devices to distributed GPU clusters. This paper describes both the API design and the system implementation of MXNet, and explains how embedding of both symbolic expression and tensor operation is handled in a unified fashion. Our preliminary experiments reveal promising results on large scale deep neural network applications using multiple GPU machines.
December 8, 2015 by hgpu
