jaxsnn.event.modules.lif

Modules

jaxsnn.event.modules.lif.lif

jaxsnn.event.modules.lif.parameters

Classes

LIFParameters(tau_syn, tau_mem, v_th, …)

Functions

jaxsnn.event.modules.lif.LIF(size: int, n_steps: int, params: jaxsnn.event.modules.lif.parameters.LIFParameters) → jaxsnn.event.types.Population

Creates a LIF layer for event-based simulation and backpropagation. Returns a generator function that builds the layer based on input connections, delays, and chosen backpropagation strategy. Supports both forward-only simulation and custom backward passes (eventprop).

Parameters

  • size – Number of neurons in the layer.

  • n_steps – Number of event steps in the simulation.

  • params – Parameters for the LIF neuron dynamics.

Returns

A Population object containing the generator and parameters.