jaxsnn.discrete.leaky_integrate

Classes

LIParameters(tau_syn, tau_mem, v_leak)
LIState(v, i)

Functions

jaxsnn.discrete.leaky_integrate.LI(out_dim, scale_in=0.2)

Layer constructor function for a li (leaky-integrated) layer.

jaxsnn.discrete.leaky_integrate.li_feed_forward_step(init, spikes: jax.Array, params: jaxsnn.base.params.LIParameters = LIParameters(tau_syn=0.005, tau_mem=0.01, v_leak=0.0), dt: float = 0.001)