snntorch icon indicating copy to clipboard operation
snntorch copied to clipboard

Published 20 hours ago • verified publisher icon jeshraghian

Faster Training Using Triton

Open korbexmachina opened this issue 10 months ago • 2 comments

Enhancement

Use the Triton compiler from OpenAI to accelerate model training.

korbexmachina avatar Apr 11 '24 21:04 korbexmachina

I would like to be assigned to implementing this enhancement.

korbexmachina avatar Apr 11 '24 21:04 korbexmachina

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

jeshraghian avatar Apr 11 '24 22:04 jeshraghian