-
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 software comparisons out there:
Here, I compare free ML/deep-learning libraries/frameworks 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/multi-node?).
- 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,multi-node | ? |
mxnet | py/ju | ? | yes | ? | openmp,cuda,multi-node | ? |
scikit-learn | ||||||
theano | py | ? | yes | ? | openmp,cuda,(dev-opencl),multi-node | ? |
[keras]https://keras.io/ | py | ? | yes | ? | theano-backend,multi-node | ? |
torch | ||||||
CNTK | ||||||