This paper introduces Inferno, a software library built on top of PyTorch that is designed to meet distinctive challenges of using spiking neural networks (SNNs) for machine learning tasks. We describe the architecture of Inferno and key differentiators that make it uniquely well-suited to these tasks. We show how Inferno supports trainable heterogeneous delays on both CPUs and GPUs, and how Inferno enables a "write once, apply everywhere" development methodology for novel models and techniques. We compare Inferno's performance to BindsNET, a library aimed at machine learning with SNNs, and Brian2/Brian2CUDA which is popular in neuroscience. Among several examples, we show how the design decisions made by Inferno facilitate easily implementing the new methods of Nadafian and Ganjtabesh in delay learning with spike-timing dependent plasticity.
@article{arxiv.2409.11567,
title = {Inferno: An Extensible Framework for Spiking Neural Networks},
author = {Marissa Dominijanni},
journal= {arXiv preprint arXiv:2409.11567},
year = {2024}
}
Comments
Source code is available at https://github.com/mdominijanni/inferno and documentation can be viewed at https://docs.inferno-ai.dev/ in a browsable form