jaxsnn.event.root.next_finder

Classes

LIFState

alias of jaxsnn.base.types.LIState

Functions

jaxsnn.event.root.next_finder.next_event(solver: Callable, neuron_state: jaxsnn.base.types.LIState, time: float, t_max: float) → jax.Array

Wrapper a root solver to provide a cleaner API for returning next event

Args:

solver (Callable): The actual root solver neuron_state (LIFState): The state of the neurons time (float): Current time t_max (float): Maximum time of the simulation

Returns:

Spike: Spike which will occur next