jaxsnn.event.to_nir_data

Translate spikes from a jaxsnn spike representation to NIRGraphData.

Classes

EventData(idx, time, n_neurons, t_max)

Event-based data represented as a list of spike indices and spike times.

EventPropSpike(time, idx, current)

NIRGraphData(nodes, …)

Dictionary of NIRNodeData where each entry represents a NIRNode of a corresponding NIRGraph with its observables.

NIRNodeData(observables)

Dictionary of EventData or TimeGriddedData where each entry represents an observable (e.g., spikes, voltages) of a corresponding NIRNode

Functions

jaxsnn.event.to_nir_data.to_nir_data(jaxsnn_dict: Dict[str, jaxsnn.event.types.EventPropSpike], jaxsnn_model) → nir.data_ir.graph.NIRGraphData
Parameters

  • jaxsnn_dict – Dictionary of Spike objects where each entry represents the spikes for a corresponding input node of the jaxsnn_model. Empty events in jaxsnn are encoded by idx = -1 and time = np.inf.

  • jaxsnn_model – A tuple of (init_fn, apply_fn).