深度學習框架:MXNet

jopen 9年前發布 | 35K 次閱讀 MXNet 機器學習

MXNet是深學習框架,設計效率和靈活性。


cxxnet借鑒了很多caffe的思想。相比之下,cxxnet在實現上更加干凈,例如依賴很少,通過mshadow的模板化使得gpu和cpu代碼只用寫一份,分布式接口也很干凈。

mxnet是cxxnet的下一代,目前實現了cxxnet所有功能,但借鑒了minerva/torch7/theano,加入更多新的功能。
  1. ndarray編程接口,類似matlab/numpy.ndarray/torch.tensor。獨有優勢在于通過背后的engine可以在性能上和內存使用上更優
  2. symbolic接口。這個可以使得快速構建一個神經網絡,和自動求導。
  3. 更多binding 目前支持比較好的是python,馬上會有julia和R
  4. 更加方便的多卡和多機運行
  5. 性能上更優。目前mxnet比cxxnet快40%,而且gpu內存使用少了一半。
目前mxnet還在快速發展中。這個月的主要方向有三,更多的binding,更好的文檔,和更多的應用(language model、語音,機器翻譯,視頻)。

特性

  • Open sourced design note on useful insights that can re-used by general DL projects.
  • Flexible configuration, for arbitrary computation graph.
  • Mix and Maximize good flavours of programmingto maximize flexibility and efficiency.
  • Lightweight, memory efficient and portable to smart devices.
  • Scales up to multi GPUs and distributed setting with auto parallelism.
  • Support python, R, C++, Julia,
  • Cloud friendly, and directly compatible with S3, HDFS, AZure

項目主頁:http://www.baiduhome.net/lib/view/home/1447421082397

 本文由用戶 jopen 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!