Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
Python C++ Perl Jupyter Notebook Scala Cuda Other
Permalink
Failed to load latest commit information.
.github Update PR Template (#9919) Mar 17, 2018
3rdparty [MXNET-626] Add while_loop (#11566) Jul 19, 2018
R-package bump up version number to 1.3.0 to make nightly build to build with r… May 27, 2018
amalgamation [3rdparty] introduce tvm (#11130) Jun 3, 2018
benchmark/python [MXNET-626] Add while_loop (#11566) Jul 19, 2018
ci Refactor armv8 builds and add to CI (#11579) Jul 19, 2018
cmake Add path_suffixes for finding cudnn.lib (#11513) Jul 6, 2018
contrib/clojure-package Update README.md (#11677) Jul 13, 2018
cpp-package [MXNET-560] Add temperature parameter in Softmax operator (#11466) Jul 19, 2018
docker [Perl] Sparse feature. (#9988) Mar 6, 2018
docs updated docs and hashes for vgg16,16_bn,19,19_bn (#11830) Jul 21, 2018
example New ecosystem website page (#11528) Jul 20, 2018
include/mxnet Fix flaky test test_gluon.test_hybrid_static_memory_switching (#11577) Jul 11, 2018
make [MXNET-472] ccache for docker builds (#11151) Jun 8, 2018
matlab Adding Apache Header to .m, .cfg, R and .mk files (#9499) Jan 22, 2018
perl-package MXNET-650 [Perl] Gluon ModelZoo, Gluon examples (#11642) Jul 13, 2018
plugin add input argument in warpctc layer (#11167) Jun 6, 2018
python updated docs and hashes for vgg16,16_bn,19,19_bn (#11830) Jul 21, 2018
scala-package [MXNET-600][Scala] NDArray auto-collector (#11751) Jul 19, 2018
setup-utils Update expected result in osx python install script (#10842) May 10, 2018
src [MXNET-489] MKLDNN Pool test (#11608) Jul 20, 2018
tests Flaky: Disabled broadcast_mod unit test (#11845) Jul 21, 2018
tools Some Python 3 fixes in ./tools again (#11757) Jul 18, 2018
.codecov.yml [MXNET-561] Add test coverage reporting (#11344) Jun 21, 2018
.gitattributes [R] To ignore R-pkg when releasing on github (#7007) Jul 13, 2017
.gitignore Clojure Contrib Package (#11205) Jul 1, 2018
.gitmodules [3rdparty] introduce tvm (#11130) Jun 3, 2018
.mxnet_root CI docker revamp; Add Jetson, Raspberry and CentOS 7 build [MXNET-42]… Mar 9, 2018
.travis.yml Add h5py support to NDArrayIter (#6790) Jul 18, 2017
CMakeLists.txt CMake 3.12 will not detect a variable as empty string (#11584) Jul 8, 2018
CODEOWNERS Add CODEOWNERS (#11598) Jul 7, 2018
CONTRIBUTORS.md [MXNET-560] Add temperature parameter in Softmax operator (#11466) Jul 19, 2018
DISCLAIMER Add DISCLAIMER and lxn2 GPG keys (#7344) Aug 5, 2017
Jenkinsfile Refactor armv8 builds and add to CI (#11579) Jul 19, 2018
KEYS Add gpg keys (#10701) Apr 30, 2018
LICENSE [MXNET-533] MXNet-ONNX export (#11213) Jun 25, 2018
MKL_README.md Add Windows MKLDNN Building Instruction (#10613) May 18, 2018
Makefile [MXNET-531] Integration Test for Scala (#11596) Jul 12, 2018
NEWS.md NEWS and README update to master (#11017) May 23, 2018
NOTICE Update NOTICE (#9706) Feb 6, 2018
README.md Updating readme to fix broken docs status (#11102) Jun 1, 2018
appveyor.yml License Adds - some more (#9559) Jan 26, 2018
mkldnn.mk fix. (#11471) Jun 28, 2018
readthedocs.yml [docs] add favicon and fix index html title Mar 25, 2016
snap.python Add snapcraft packaging (#4852) Mar 23, 2017
snapcraft.yaml bump up version number to 1.3.0 to make nightly build to build with r… May 27, 2018

README.md


Apache MXNet (incubating) for Deep Learning

Master Docs License
Build Status Documentation Status GitHub license

banner

Apache MXNet (incubating) is a deep learning framework designed for both efficiency and flexibility. It allows you to mix symbolic and imperative programming to maximize efficiency and productivity. At its core, MXNet contains a dynamic dependency scheduler that automatically parallelizes both symbolic and imperative operations on the fly. A graph optimization layer on top of that makes symbolic execution fast and memory efficient. MXNet is portable and lightweight, scaling effectively to multiple GPUs and multiple machines.

MXNet is also more than a deep learning project. It is also a collection of blue prints and guidelines for building deep learning systems, and interesting insights of DL systems for hackers.

Join the chat at https://gitter.im/dmlc/mxnet

What's New

Contents

Features

  • Design notes providing useful insights that can re-used by other DL projects
  • Flexible configuration for arbitrary computation graph
  • Mix and match imperative and symbolic programming to maximize flexibility and efficiency
  • Lightweight, memory efficient and portable to smart devices
  • Scales up to multi GPUs and distributed setting with auto parallelism
  • Support for Python, R, Scala, C++ and Julia
  • Cloud-friendly and directly compatible with S3, HDFS, and Azure

Ask Questions

License

Licensed under an Apache-2.0 license.

Reference Paper

Tianqi Chen, Mu Li, Yutian Li, Min Lin, Naiyan Wang, Minjie Wang, Tianjun Xiao, Bing Xu, Chiyuan Zhang, and Zheng Zhang. MXNet: A Flexible and Efficient Machine Learning Library for Heterogeneous Distributed Systems. In Neural Information Processing Systems, Workshop on Machine Learning Systems, 2015

History

MXNet emerged from a collaboration by the authors of cxxnet, minerva, and purine2. The project reflects what we have learned from the past projects. MXNet combines aspects of each of these projects to achieve flexibility, speed, and memory efficiency.