jaxsnn.event.from_nir_data

Translate NIRGraphData to a jaxsnn-compatible spike representation.

Classes

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

TimeGriddedData(data, dt)

Either boolean entries indicate whether a binary event is present at a particular time step, or a real-valued signal provides the measurement of a quantity (e.g.

Functions

jaxsnn.event.from_nir_data.from_nir_data(nir_graph_data: nir.data_ir.graph.NIRGraphData, jaxsnn_model, observables=('spikes')) → Dict[str, jaxsnn.event.types.EventPropSpike]

Convert NIRGraphData to a dict of EventPropSpikes (jax-snn representation)

Parameters

  • nir_graph_data – NIRGraphData to be converted.

  • jaxsnn_model – jaxsnn model tuple (init, apply).

  • observables – Observables to be converted, by default (‘spikes’,)