Skip to content

alpha-1 release

Pre-release
Pre-release
Compare
Choose a tag to compare
@jeshraghian jeshraghian released this 11 Feb 08:27
· 1098 commits to master since this release
v0.1.2
400ad46

The first functional iteration of snnTorch!

What's new?

snntorch
The workhorse of the package.
All neuron models are integrated here, and a default Heaviside gradient is used to override the non-differentiability with conventional autograd methods in PyTorch.

  • Stein's neuron model
  • SRM0 neuron model
  • firing inhibition, thanks to @xxwang1
  • hidden states can optionally be initialized as instance variables if the user wants to just use a built-in backprop method

snntorch.backprop

  • Backprop through time (BPTT)
  • Truncated backprop through time (TBPTT)
  • Real-time recurrent learning (RTRL)

snntorch.spikegen

  • Poisson spike train generator
  • Rate coding
  • Latency coding

snntorch.surrogate

  • FastSigmoid
  • Sigmoid
  • Spike Rate Escape

snntorch.spikeplot

  • Raster plots
  • Feature map animator
  • Spike count animator

snntorch.utils

  • Data split
  • Data reduction

Plans for alpha-2

  • delta & delta-sigma spike generators for snntorch.spikegen
  • Simplified Stein's model (reduce hidden states from 2 to 1)
  • More surrogate and backprop methods
  • add more tests
Assets 2
Loading