bamsumit/slayerPytorch

PyTorch implementation of SLAYER for training Spiking Neural Networks

What it solves

It addresses the challenge of training multilayer spiking neural networks (SNNs) by enabling the learning of synaptic weights and axonal delay parameters through backpropagation.

How it works

It is a PyTorch port of the SLAYER (Spike Layer Error Reassignment in Time) framework. The system natively handles multiple spikes per layer and implements error backpropagation through the layers to optimize network parameters.

Who it’s for

Researchers and developers working with spiking neural networks who want to implement and train these models using PyTorch and CUDA-enabled GPUs.

Highlights

  • PyTorch implementation of the SLAYER framework.
  • Supports learning of both synaptic weights and axonal delays.
  • Natively handles multiple spikes per layer.
  • Includes example implementations for Multi-Layer Perceptrons (MLP) and Convolutional Neural Networks (CNN).

Related

  • Project
    BindsNET/bindsnet

    A PyTorch-based library for simulating spiking neural networks (SNNs) to develop biologically inspired machine learning and reinforcement learning algorithms.

  • Project
    Lightning-AI/pytorch-lightning

    A deep learning framework that abstracts away the engineering boilerplate of PyTorch, enabling researchers to pretrain and finetune AI models at scale across various hardware.

  • Project
    pytorch/pytorch

    A Python-based machine learning framework that provides GPU-accelerated tensor computation and a dynamic autograd system for building deep neural networks.

  • Project
    TorchSpatiotemporal/tsl

    A PyTorch-based library for neural spatiotemporal forecasting that provides a unified framework for data preprocessing, model prototyping, and Graph Neural Network implementation.

  • Project
    fangwei123456/spikingjelly

    SpikingJelly is a PyTorch‑based library for building, training, and deploying spiking neural networks, offering multiple performance back‑ends, large‑scale training tools, neuromorphic dataset loaders, and export to hardware formats.