Kernel.init_state()# abstract Kernel.init_state(prng_key, model_state)[source]# Creates the initial kernel state. Return type: Any