Skip to content

Machine Learning library comparisons for testing brain architectures

Aditya Gilra edited this page Feb 9, 2017 · 15 revisions

There are already nice ML comparisons out there:

  1. https://en.wikipedia.org/wiki/Comparison_of_deep_learning_software

Here, I compare free ML libraries for suitability of use in archibrain.
Particularly, we want
0. python / julia support

  1. arbitrary even spiking neurons
  2. auto-differentiation of arbitrary functions
  3. ease of implementing different learning algos (not just backprop) - almost Brian style.
  4. can use GPUs (and CPUs) across multiple nodes (openmp/cuda/opencl?).
  5. ease of implementing both RL and supervised learning tasks.
library py/ju? arb nrns? auto-diff? arb learn algo? multi-node GPU/CPU tasks
tensorflow py ? yes ? cuda ?
mxnet py/ju ? yes ? openmp,cuda ?
scikit-learn
theano py ? yes ? openmp,cuda,(dev-opencl) ?
keras py ? yes ? (theano backend) ?
torch
CNTK
Clone this wiki locally