NUTSKernel.init_state()

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).