jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python

https://snntorch.readthedocs.io/en/latest/

MIT License

1.37k stars 228 forks source link

Open korbexmachina opened 7 months ago

korbexmachina commented 7 months ago

Enhancement

Use the Triton compiler from OpenAI to accelerate model training.

korbexmachina commented 7 months ago

I would like to be assigned to implementing this enhancement.

jeshraghian commented 7 months ago

I recommend starting with the snn.Leaky neuron model; that's perhaps the simplest/most common model used.