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