Kernel.init_state()#

abstract Kernel.init_state(prng_key, model_state)[source]#

Creates the initial kernel state.

Return type

Any