HMCKernel.init_state()# HMCKernel.init_state(prng_key, model_state)[source]# Initializes the kernel state with an identity inverse mass matrix and a reasonable step size (unless explicit arguments were provided by the user).