jaxsnn.event.modules.leaky_integrate

Classes

`LIFParameters`(tau_syn, tau_mem, v_th, …)
`LIFState`(V, I)
`Spike`(time, idx)
`partial`	partial(func, args, *keywords) - new function with partial application of the given arguments and keywords.

Functions

jaxsnn.event.modules.leaky_integrate.LeakyIntegrator(n_hidden: int, t_max: float, params: jaxsnn.base.params.LIFParameters, mean: float = 0.5, std: float = 2.0, time_steps: int = 20)

jaxsnn.event.modules.leaky_integrate.kernel_fn(kernel, time, spike_time)

jaxsnn.event.modules.leaky_integrate.leaky_integrator(kernel: jax.Array, time_steps: jax.Array, weights: jax.Array, spikes: jaxsnn.event.types.Spike) → jaxsnn.event.types.LIFState : Vectorized version of li_cell. Takes similar arguments as li_cell but with additional array axes over which li_cell is mapped.

jaxsnn.event.modules.leaky_integrate.li_cell(kernel: jax.Array, time_steps: jax.Array, weights: jax.Array, spikes: jaxsnn.event.types.Spike) → jaxsnn.event.types.LIFState 

jaxsnn.event.modules.leaky_integrate.superposition(kernel, spike_time, initial_state, time)