Perpetual HMC tags Sampling This is more of a project than a note but OK def kernel(state, rng_key): new_state =