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
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
Simple, just type
pip install git+https://github.com/MarkusAI/rA9
MacOS does not support NVIDIA CUDA.
Simple, just type
pip install git+https://github.com/MarkusAI/rA9
You need to setup JAX before installing the rA9 as GPU-dedicated and install rA9 as following:
pip install git+https://github.com/MarkusAI/rA9
Check LeNet.py in example