numba implementation of LIF

lif.py

@jit(double[:,:2](double[:], double[:], double, double, double))
def lif_model(ge, gi, dt=dt, vreset=0, vthresh=-40*mV):
    v = vreset
    mem_pot = np.zeros((len(ge), 2))
    Ee, Ei = 0, -100
    El = -60
    gl = 1/60
    C = 250
    for i in range(len(ge)):
        dv = (-gl*(v-El) - ge[i]*(v-Ee) - gi[i]*(v-Ei))*dt/C
        v = v + dv
        if (v>vthresh):
            v = vreset
            mem_pot[i, 1]=1
        mem_pot[i, 0] = v
    return  mem_pot