jaxsnn

Modules

jaxsnn.base
jaxsnn.discrete
jaxsnn.event

Classes

ConversionConfig(t_max, n_spikes, int], …)

Configuration for the conversion from NIR to jaxsnn.

Functions

jaxsnn.from_nir(graph: nir.ir.graph.NIRGraph, config: jaxsnn.event.from_nir.ConversionConfig)

Convert NIRGraph to jax-snn representation (init_fn, apply_fn)

Restrictions for NIRGraph: - Only linear feed-forward SNNs are supported - CubaLIF and Linear layers are supported - Affine layers with bias==0 are currently supported - In terms of parameters, only homogeneous layers are supported - The analytical solver is only supported for non-external inputs

Example: `python nir_graph = nir.NIRGraph(...) cfg = jaxsnn.ConversionConfig(...) init, apply = jaxsnn.from_nir(nir_graph, cfg) `

jaxsnn.get_logger(name: str)