NUTSKernel.init_state()# NUTSKernel.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).