jaxsnn.discrete.modules.li

Classes

DenseData	alias of jax.Array
LIParameters(tau_syn, tau_mem, v_leak)
LIState(V, I)
Parameter	alias of jax.Array
Population(generator, …)	A population module for the discrete paradigm.

Functions

jaxsnn.discrete.modules.li.LI(size: int, params: jaxsnn.discrete.functional.li.LIParameters = LIParameters(tau_syn=0.005, tau_mem=0.01, v_leak=0.0)) → jaxsnn.discrete.types.Population

Layer constructor function for a leaky-integrator layer.

Parameters

• size – Number of neurons in the layer.

• params – Parameters for the LI neuron model.

Returns

A Population object containing the layer definition.

jaxsnn.discrete.modules.li.li_step(inputs: Dict[str, jax.Array], state: jaxsnn.discrete.functional.li.LIState, parameters: Optional[jax.Array, None], v_leak: float, tau_mem: float, tau_syn: float, dt: float) → Tuple[jaxsnn.discrete.functional.li.LIState, jax.Array]

Euler step of a leaky integrate neuron.

Parameters

• inputs – Dictionary of input currents

• state – Current neuron state

• parameters – Optional learnable parameters

• v_leak – Leak potential

• tau_mem – Membrane time constant

• tau_syn – Synaptic time constant

• dt – Time step size

Returns

Tuple of updated neuron state and membrane potential