RWKernel.init_state()# RWKernel.init_state(prng_key, model_state)[source]# Initializes the kernel state.