jeshraghian/snntorch

Faster Training Using Triton

korbexmachina opened this issue · 2 comments

korbexmachina commented

Enhancement

Use the Triton compiler from OpenAI to accelerate model training.

korbexmachina commented

I would like to be assigned to implementing this enhancement.

jeshraghian commented

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

  • 👍2