jaxsnn.event.utils

Modules

jaxsnn.event.utils.branch_free([enabled])	Context manager to temporarily switch branch-free mode for custom_lax.cond/switch.
jaxsnn.event.utils.from_nir	Conversion of a NIR graph to jaxsnn model
jaxsnn.event.utils.from_nir_data	Translate NIRGraphData to a jaxsnn-compatible spike representation.
jaxsnn.event.utils.time_it(timed_function, *args)
jaxsnn.event.utils.to_nir	Conversion of a jaxsnn model to a NIR graph
jaxsnn.event.utils.to_nir_data	Translate spikes from a jaxsnn spike representation to NIRGraphData.

Functions

jaxsnn.event.utils.branch_free(enabled: bool = True)

Context manager to temporarily switch branch-free mode for custom_lax.cond/switch.

jaxsnn.event.utils.is_branch_free() → bool

jaxsnn.event.utils.time_it(timed_function: Callable, *args) → Tuple[Any, float]