jaxsnn.event.utils.from_nir_data

Translate NIRGraphData to a jaxsnn-compatible spike representation.

Classes

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
Spike(time, idx, current, layer_idx, internal)
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.
Topology(t_max, backprop_method, mock, …)	Represents a spiking neural network (SNN) topology as a directed graph of layers.

Functions

jaxsnn.event.utils.from_nir_data.from_nir_data(nir_graph_data: nir.data_ir.graph.NIRGraphData, topology: jaxsnn.event.topology.Topology, observables=('spikes')) → Dict[str, jaxsnn.event.types.Spike]

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

Parameters

• nir_graph_data – NIRGraphData to be converted.

• topology – jaxsnn Topology object.

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