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

Installation

MacOS

CPU

Simple, just type

pip install git+https://github.com/MarkusAI/rA9

GPU

MacOS does not support NVIDIA CUDA.

Linux

CPU

Simple, just type

pip install git+https://github.com/MarkusAI/rA9

GPU

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

Example

Check LeNet.py in example