jaxsnn.event.types.PopulationFunctions

class jaxsnn.event.types.PopulationFunctions(init: InitFn, step: EventStepFn, state: EventStateFn, event: EventFn, adjoint_step: Optional[AdjointStepFn] = None, hx_module: Optional[GenModuleFn] = None)

Bases: object

Function container for event population modules

__init__(init: InitFn, step: EventStepFn, state: EventStateFn, event: EventFn, adjoint_step: Optional[AdjointStepFn] = None, hx_module: Optional[GenModuleFn] = None) → None: Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(init, step, state, event[, …])

Initialize self.

Attributes

`adjoint_step`
`hx_module`

adjoint_step: Optional[AdjointStepFn] = None

event: EventFn

hx_module: Optional[GenModuleFn] = None

init: InitFn

state: EventStateFn

step: EventStepFn