Slax: A Composable JAX Library for Rapid and Flexible Prototyping of Spiking Neural Networks

Thomas M. Summe, Siddharth Joshi

Recent advances to algorithms for training spiking neural networks (SNNs) often leverage their unique dynamics. While backpropagation through time (BPTT) with surrogate gradients d…