jaxsnn.event.transition

Classes

LIFParameters(tau_syn, tau_mem, v_th, …)
StepState(neuron_state, spike_times, …)
WeightInput(input,)
WeightRecurrent(input, recurrent)

Functions

jaxsnn.event.transition.transition_with_recurrence(params: jaxsnn.base.params.LIFParameters, state: jaxsnn.event.types.StepState, weights: Union[jaxsnn.event.types.WeightInput, jaxsnn.event.types.WeightRecurrent], spike_mask: jax.Array, recurrent_spike: bool, prev_layer_start: int) → jaxsnn.event.types.StepState

jaxsnn.event.transition.transition_without_recurrence(params: jaxsnn.base.params.LIFParameters, state: jaxsnn.event.types.StepState, weights: jaxsnn.event.types.WeightInput, spike_mask: jax.Array, recurrent_spike: bool, prev_layer_start: int) → jaxsnn.event.types.StepState