spark.core.tracers

Classes

BaseTracer	Base Tracer class
Tracer	Multipurpose exponential tracer.
RDTracer	Rise-Decay Tracer.
RFSTracer	Rise-Fast-Slow Tracer
RUTracer	Resource-Usage tracer for STP (Short Term Plasticity).

Module Contents

class spark.core.tracers.BaseTracer(shape, seed=None, dtype=jnp.float16, dt=1.0, **kwargs)

Bases: flax.nnx.Module, abc.ABC

Base Tracer class

Parameters:

• shape (tuple[int, Ellipsis])

• seed (int | None)

• dtype (jax.typing.DTypeLike)

• dt (float)

shape

rng

abstractmethod reset()

Return type:

None

abstractmethod masked_reset(mask)

Return type:

None

property value: jax.Array

Abstractmethod:

Return type:

jax.Array

__call__(x, **kwargs)

Parameters:

x (jax.Array)

Return type:

jax.Array

class spark.core.tracers.Tracer(shape, tau, scale=1, base=0, **kwargs)

Bases: BaseTracer

Multipurpose exponential tracer.

Parameters:

• shape (tuple[int, Ellipsis])

• tau (jax.Array | float)

• scale (jax.Array | float)

• base (jax.Array | float)

scale

base

decay

trace

reset()

Return type:

None

masked_reset(mask)

Return type:

None

property value: jax.Array

Return type:

jax.Array

class spark.core.tracers.RDTracer(shape, tau_rise, tau_decay, scale_rise=1, scale_decay=1, base_rise=0, base_decay=0, **kwargs)

Bases: BaseTracer

Rise-Decay Tracer.

Multipurpose double exponential tracer.

Parameters:

• shape (tuple[int, Ellipsis])

• tau_rise (jax.Array | float)

• tau_decay (jax.Array | float)

• scale_rise (jax.Array | float)

• scale_decay (jax.Array | float)

• base_rise (jax.Array | float)

• base_decay (jax.Array | float)

tracer_rise

tracer_decay

reset()

Return type:

None

masked_reset(mask)

Return type:

None

property value: jax.Array

Return type:

jax.Array

class spark.core.tracers.RFSTracer(shape, alpha, tau_rise, tau_fast_decay, tau_slow_decay, scale_rise=1, scale_fast_decay=1, scale_slow_decay=1, base_rise=0, base_fast_decay=0, base_slow_decay=0, **kwargs)

Bases: BaseTracer

Rise-Fast-Slow Tracer

Multipurpose triple exponential tracer.

Parameters:

• shape (tuple[int, Ellipsis])

• alpha (jax.Array | float)

• tau_rise (jax.Array | float)

• tau_fast_decay (jax.Array | float)

• tau_slow_decay (jax.Array | float)

• scale_rise (jax.Array | float)

• scale_fast_decay (jax.Array | float)

• scale_slow_decay (jax.Array | float)

• base_rise (jax.Array | float)

• base_fast_decay (jax.Array | float)

• base_slow_decay (jax.Array | float)

tracer_rise_fast

tracer_rise_slow

alpha

reset()

Return type:

None

masked_reset(mask)

Return type:

None

property value: jax.Array

Return type:

jax.Array

class spark.core.tracers.RUTracer(shape, r_tau, u_tau, u_scale, **kwargs)

Bases: BaseTracer

Resource-Usage tracer for STP (Short Term Plasticity).

Parameters:

• shape (tuple[int, Ellipsis])

• r_tau (jax.Array | float)

• u_tau (jax.Array | float)

• u_scale (jax.Array | float)

r_tracer

u_tracer

reset()

Return type:

None

masked_reset(mask)

Return type:

None

property value: jax.Array

Return type:

jax.Array