jeshraghian/snntorch
snnTorch is a Python package that extends PyTorch with spiking neural network models for GPU-accelerated deep and online learning.

Velocity · 7d
+1.0
★ / day
Trend
→steady
star history
snnTorch is a Python package for performing gradient-based learning with spiking neural networks. It extends PyTorch’s GPU-accelerated tensor computation to networks of spiking neurons, integrating pre-designed spiking neuron models as recurrent activation units that can be trained using standard PyTorch optimizers and workflows.