-
Notifications
You must be signed in to change notification settings - Fork 3
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:
Here, I compare free ML libraries for suitability of use in archibrain.
Particularly, we want
0. python / julia support
- arbitrary even spiking neurons
- auto-differentiation of arbitrary functions
- ease of implementing different learning algos (not just backprop) - almost Brian style.
- can use GPUs (and CPUs) across multiple nodes (openmp/cuda/opencl?).
- 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 | ||||||