PyTorch STDP implementation for spiking neural networks. Does not include LIF neuron behavior. Weight changing rule is from Gerstner et al. 2014
Gerstner, W., Kistler, W. M., Naud, R., & Paninski, L. (2014). Neuronal dynamics: From single neurons to networks and models of cognition. Cambridge University Press.
STDP.py has the learning rule, main.py has an example model with fake input spikes and spike values.