nengo / pytorch-spiking

Spiking neuron integration for PyTorch
https://www.nengo.ai/pytorch-spiking/
Other
38 stars 6 forks source link

Neuron models #7

Open LortKrassbeter opened 3 years ago

LortKrassbeter commented 3 years ago

Hi,

it seems to me that currently this library implements a kind of integrate and fire neuron in the SpikingActivation module. I was wondering if there are plans for modules that implement more complex neuron models like LIF and Izhikevich

Best

drasmuss commented 3 years ago

We don't have any plans for that currently, but we would certainly welcome PRs. You may also be interested in NengoDL, which has support for many more neuron types (but it's implemented in TensorFlow rather than PyTorch).