RWKernel.end_epoch()# RWKernel.end_epoch(prng_key, kernel_state, model_state, epoch)[source]# Sets the step size as found by the dual averaging algorithm. Return type: RWKernelState