jaxsnn.event.stepping.step_multi_layer

Classes

Carry(parameters, spikes, external_spikes, …)

Step(parameters, spikes, external_spikes, …)

Functions

jaxsnn.event.stepping.step_multi_layer.multi_layer_step(step_fns: Dict[str, Callable], nodes: List[str], node_index_mapping: Dict[str, int], carry: jaxsnn.event.types.Carry, step_idx: int) → Tuple[jaxsnn.event.types.Carry, int]

Perform one simulation step across multiple neuron layers.

Iterates through all layers in the strongly connected component (SCC), applies their respective step functions, and updates spikes, states, queue heads, and queue indices accordingly.

Parameters

  • step_fns – Dictionary of step functions for each layer, keyed by node name.

  • nodes – List of node names in the SCC to process.

  • node_index_mapping – Dictionary mapping node names to their indices.

  • carry – Tuple (parameters, spikes, states, queue_heads, queue_indices).

  • step_idx – Current step index.

Returns

Updated carry tuple and an integer placeholder (always 0).