HMCKernel.start_epoch()# HMCKernel.start_epoch(prng_key, kernel_state, model_state, epoch)[source]# Resets the state of the dual averaging algorithm. Return type: HMCKernelState