It seems like the most straightforward way to handle this would be for the PPL to to allow you to carry around your sampler/step method, initialize it, tune it, draw samples from it etc. But that doesn’t really conform to the current pymc API nor it’s design and reliance on the model context. But it may be possible by accessing non-obvious parts of the API.