This looks pretty good at a first glance. So your main proposal is subclassing tfp.mcmc so we can interact with it a bit more flexible?
This looks pretty good at a first glance. So your main proposal is subclassing tfp.mcmc so we can interact with it a bit more flexible?