I see. Yeah that part of the code is not very exposed. When you say peek while sample do you mean peeking the parameters in the sampler? Like kernel size etc? Maybe it is easier to cherry pick the code from pm.sample and call pm.sampling._iter_sample directly.