Shypd.aiThe largest open index of AI toolssnntorch is a Python package for deep and online learning with spiking neural networks (SNNs). It extends PyTorch capabilities for gradient-based learning with SNNs, leveraging GPU acceleration.
Trending
About
snntorch is a Python package designed for deep and online learning with spiking neural networks (SNNs). It seamlessly integrates with PyTorch, utilizing its GPU-accelerated tensor computation for SNNs. The library provides pre-designed spiking neuron models that function as recurrent activation units within the PyTorch framework. snntorch is agnostic to various layer types like fully-connected or convolutional layers, making it versatile for different network architectures. It features components for spiking neuron libraries, exporting/importing to other SNN libraries via NIR, common arithmetic operations on spikes, spike generation, data conversion, and visualization tools. The design emphasizes lean requirements, enabling training on both CPU and GPU.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
