search

jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python
https://snntorch.readthedocs.io/en/latest/
MIT License
1.17k stars 202 forks source link

Faster Training Using Triton #312

Open korbexmachina opened 2 months ago

korbexmachina commented 2 months ago

Enhancement

Use the Triton compiler from OpenAI to accelerate model training.

korbexmachina commented 2 months ago

I would like to be assigned to implementing this enhancement.

jeshraghian commented 2 months ago

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