jaxsnn.event.loss

Classes

LIFState(V, I)
Spike(time, idx, current, layer_idx, internal)

Functions

jaxsnn.event.loss.first_spike(spikes: jaxsnn.event.types.Spike, n_outputs: int) → jax.Array

jaxsnn.event.loss.max_over_time(output: jaxsnn.event.states.LIFState) → jax.Array

jaxsnn.event.loss.mse_loss(first_spikes: jax.Array, target: jax.Array, tau_mem: float) → jax.Array

jaxsnn.event.loss.nll_loss(output: jax.Array, targets: jax.Array) → jax.Array

jaxsnn.event.loss.target_time_loss(first_spikes: jax.Array, target: jax.Array, tau_mem: float) → jax.Array

jaxsnn.event.loss.ttfs_loss(first_spikes: jax.Array, target: jax.Array, tau_mem: float) → jax.Array