jaxsnn.event.root.next_finder

Classes

LIFState(V, I)

Functions

jaxsnn.event.root.next_finder.next_event(solver: Callable, neuron_state: jaxsnn.event.types.LIFState, 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