KernelSequence.init_states()#

KernelSequence.init_states(prng_key, model_state)[source]#

Creates the initial kernel state.

Return type:

list[Any]