Code for training probabilistic binary and WTA SNNs using PyTorch. Part of this code has been used for the following works:
N. Skatchkovsky, H. Jang, and O. Simeone, Federated Neuromorphic Learning of Spiking Neural Networks for Low-Power Edge Intelligence, accepted to IEEE International Conference on Acoustics, Speech, and Signal Processing (ICASSP), 2020. https://arxiv.org/abs/1910.09594
H. Jang, N. Skatchkovsky, and O. Simeone, VOWEL: A Local Online Learning Rule for Recurrent Networks of Probabilistic Spiking Winner-Take-All Circuits, submitted for conference publication. https://arxiv.org/abs/2004.09416