Nice! Similar to what spiking neural network people have been dealing with. E.g. https://github.com/fzenke/spytorch