I will need a bit more time to review the code. As of the proposal, I suggest removing the progress bar part, but keep the compound/Gibbs step.
Ideally, we get a version that autogenerate state_fn for RWMH (so we can handle model that mixed with discrete dtype), and a metropolis-within-gibbs TransitionKernel (ie. compound step)