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:

Here, I compare ML libraries for suitability of use in archibrain.
Particularly, we want

  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.
  5. ease of implementing both RL and supervised learning tasks.

|library|arb nrns?|auto-diff?|arb learn algo?|multi-node GPU/CPU|tasks|
|---|---|---|---|---|

Clone this wiki locally