rA9
Spiking Neural Network Library Based on JAX and referencing codes from bintorch
The learning algorithm of this library is spike-based backpropagation proposed from Enabling Spike-Based Backpropagation for Training Deep Neural Network Architectures
Compatitable Operating Systems
Only supports Linux and MacOS, because of the dependency of JAX and they try to fix it. But you can run this library in WSL for running this library in windows.
Installation
MacOS
CPU
Simple, just type
pip install rA9
GPU
MacOS does not support Nvidia CUDA accelerating.
Linux
CPU
Simple, just type
pip install rA9
GPU
You need to setup JAX before installing the rA9 as GPU-dedicated and install rA9 as following:
pip install rA9
Specific
TPU
It also supports TPU in google colab
Disclaimer: This project is still in progress.