Open main menu

Wikipedia β

Comparison of deep learning software

The following table compares some of the most popular software frameworks, libraries and computer programs for deep learning.


Contents

Deep learning software by nameEdit

Software Creator Software license[a] Open source Platform Written in Interface OpenMP support OpenCL support CUDA support Automatic differentiation[1] Has pretrained models Recurrent nets Convolutional nets RBM/DBNs Parallel execution (multi node)
Caffe Berkeley Vision and Learning Center BSD license Yes Linux, Mac OS X, Windows[2] C++ Python, MATLAB Yes Under development[3] Yes Yes Yes[4] Yes Yes No ?
Caffe2 Facebook Apache 2.0 Yes Linux, Mac OS X, Windows[5] C++, Python Python, MATLAB Yes Under development[6] Yes Yes Yes[7] Yes Yes No Yes
Deeplearning4j Skymind engineering team; Deeplearning4j community; originally Adam Gibson Apache 2.0 Yes Linux, Mac OS X, Windows, Android (Cross-platform) C++, Java Java, Scala, Clojure, Python (Keras), Kotlin Yes On roadmap[8] Yes[9][10] Computational Graph Yes[11] Yes Yes Yes Yes[12]
Dlib Davis King Boost Software License Yes Cross-Platform C++ C++ Yes No Yes Yes Yes No Yes Yes Yes
Gensim
Keras François Chollet MIT license Yes Linux, Mac OS X, Windows Python Python, R Only if using Theano or MXNet as backend Under development for the Theano backend (and on roadmap for the TensorFlow backend) Yes Yes Yes[13] Yes Yes Yes Yes[14]
MatConvNet Andrea Vedaldi, Karel Lenc BSD license Yes Windows, Linux[15] (OSX via Docker on roadmap) C++ MATLAB, C++, No No Yes Yes Yes Yes Yes No Yes
Microsoft Cognitive Toolkit Microsoft Research MIT license[16] Yes Windows, Linux[15] (OSX via Docker on roadmap) C++ Python (Keras), C++, Command line,[17] BrainScript[18] (.NET on roadmap[19]) Yes[20] No Yes Yes Yes[21] Yes[22] Yes[22] No[23] Yes[24]
MXNet Distributed (Deep) Machine Learning Community Apache 2.0 Yes Linux, Mac OS X, Windows,[25][26] AWS, Android,[27] iOS, JavaScript[28] Small C++ core library C++, Python, Julia, Matlab, JavaScript, Go, R, Scala, Perl Yes On roadmap[29] Yes Yes[30] Yes[31] Yes Yes Yes Yes[32]
Neural Designer Artelnics Proprietary No Linux, Mac OS X, Windows C++ Graphical user interface Yes No No ? ? No No No ?
OpenNN Artelnics GNU LGPL Yes Cross-platform C++ C++ Yes No No ? ? No No No ?
Paddle
Pytorch
Apache SINGA Apache Incubator Apache 2.0 Yes Linux, Mac OS X, Windows C++ Python, C++, Java No Yes Yes ? Yes Yes Yes Yes Yes
TensorFlow Google Brain team Apache 2.0 Yes Linux, Mac OS X, Windows[33] C++, Python Python (Keras), C/C++, Java, Go, R[34] No On roadmap[35] but already with SYCL[36] support Yes Yes[37] Yes[38] Yes Yes Yes Yes
Theano Université de Montréal BSD license Yes Cross-platform Python Python (Keras) Yes Under development[39] Yes Yes[40][41] Through Lasagne's model zoo[42] Yes Yes Yes Yes[43]
Torch Ronan Collobert, Koray Kavukcuoglu, Clement Farabet BSD license Yes Linux, Mac OS X, Windows,[44] Android,[45] iOS C, Lua Lua, LuaJIT,[46] C, utility library for C++/OpenCL[47] Yes Third party implementations[48][49] Yes[50][51] Through Twitter's Autograd[52] Yes[53] Yes Yes Yes Yes[54]
Wolfram Mathematica Wolfram Research Proprietary No Windows, Mac OS X, Linux, Cloud computing C++ Wolfram Language No No Yes Yes Yes[55] Yes Yes Yes Yes
  1. ^ Licenses here are a summary, and are not taken to be complete statements of the licenses. Some libraries may use other libraries internally under different licenses

Related softwareEdit

See alsoEdit

ReferencesEdit

  1. ^ Atilim Gunes Baydin; Barak A. Pearlmutter; Alexey Andreyevich Radul; Jeffrey Mark Siskind (20 February 2015). "Automatic differentiation in machine learning: a survey". arXiv:1502.05767  [cs.LG]. 
  2. ^ "Microsoft/caffe". GitHub. 
  3. ^ "OpenCL Caffe". 
  4. ^ "Caffe Model Zoo". 
  5. ^ "Caffe2 Github Repo". 
  6. ^ "OpenCL Caffe". 
  7. ^ "Caffe Model Zoo". 
  8. ^ "Support for Open CL · Issue #27 · deeplearning4j/nd4j". GitHub. 
  9. ^ "N-Dimensional Scientific Computing for Java". 
  10. ^ "Comparing Top Deep Learning Frameworks". Deeplearning4j. 
  11. ^ Chris Nicholson; Adam Gibson. "Deeplearning4j Models". 
  12. ^ Deeplearning4j. "Deeplearning4j on Spark". Deeplearning4j. 
  13. ^ https://keras.io/applications/
  14. ^ Does Keras support using multiple GPUs? · Issue #2436 · fchollet/keras
  15. ^ a b "Setup CNTK on your machine". GitHub. 
  16. ^ "CNTK/LICENSE.md at master · Microsoft/CNTK · GitHub". GitHub. 
  17. ^ "CNTK usage overview". GitHub. 
  18. ^ "BrainScript Network Builder". GitHub. 
  19. ^ ".NET Support · Issue #960 · Microsoft/CNTK". GitHub. 
  20. ^ "How to train a model using multiple machines? · Issue #59 · Microsoft/CNTK". GitHub. 
  21. ^ https://github.com/Microsoft/CNTK/issues/140#issuecomment-186466820
  22. ^ a b "CNTK - Computational Network Toolkit". Microsoft Corporation. 
  23. ^ url=https://github.com/Microsoft/CNTK/issues/534
  24. ^ "Multiple GPUs and machines". Microsoft Corporation. 
  25. ^ "Releases · dmlc/mxnet". Github. 
  26. ^ "Installation Guide — mxnet documentation". Readthdocs. 
  27. ^ "MXNet Smart Device". ReadTheDocs. 
  28. ^ "MXNet.js". Github. 
  29. ^ "Support for other Device Types, OpenCL AMD GPU · Issue #621 · dmlc/mxnet". GitHub. 
  30. ^ http://mxnet.readthedocs.io/
  31. ^ "Model Gallery". GitHub. 
  32. ^ "Run MXNet on Multiple CPU/GPUs with Data Parallel". GitHub. 
  33. ^ https://developers.googleblog.com/2016/11/tensorflow-0-12-adds-support-for-windows.html
  34. ^ interface), JJ Allaire (R; RStudio; Eddelbuettel, Dirk; Golding, Nick; Tang, Yuan; Tutorials), Google Inc (Examples and (2017-05-26), tensorflow: R Interface to TensorFlow, retrieved 2017-06-14 
  35. ^ "tensorflow/roadmap.md at master · tensorflow/tensorflow · GitHub". GitHub. January 23, 2017. Retrieved May 21, 2017. 
  36. ^ "OpenCL support · Issue #22 · tensorflow/tensorflow". GitHub. 
  37. ^ https://www.tensorflow.org/
  38. ^ https://github.com/tensorflow/models
  39. ^ "Using the GPU — Theano 0.8.2 documentation". 
  40. ^ http://deeplearning.net/software/theano/library/gradient.html
  41. ^ https://groups.google.com/d/msg/theano-users/mln5g2IuBSU/gespG36Lf_QJ
  42. ^ "Recipes/modelzoo at master · Lasagne/Recipes · GitHub". GitHub. 
  43. ^ Using multiple GPUs — Theano 0.8.2 documentation
  44. ^ https://github.com/torch/torch7/wiki/Windows
  45. ^ "GitHub - soumith/torch-android: Torch-7 for Android". GitHub. 
  46. ^ "Torch7: A Matlab-like Environment for Machine Learning" (PDF). 
  47. ^ "GitHub - jonathantompson/jtorch: An OpenCL Torch Utility Library". GitHub. 
  48. ^ "Cheatsheet". GitHub. 
  49. ^ "cltorch". GitHub. 
  50. ^ "Torch CUDA backend". GitHub. 
  51. ^ "Torch CUDA backend for nn". GitHub. 
  52. ^ https://github.com/twitter/torch-autograd
  53. ^ "ModelZoo". GitHub. 
  54. ^ https://github.com/torch/torch7/wiki/Cheatsheet#distributed-computing--parallel-processing
  55. ^ http://blog.stephenwolfram.com/2017/03/the-rd-pipeline-continues-launching-version-11-1/