Help navigating implementation of new inference algorithm

Getting very close to done at this point, but one thing I’m having trouble is finding a clean way to sample the transformed variables given the transformed hyperparameters? Without the transformation, I have

x_RVs = model.observed_RVs
θ_RVs = [v for v in model.basic_RVs if model_graph.get_parents(v) == set()]
z_RVs = [v for v in model.free_RVs if v not in θ_RVs]

sample_x_z = aesara.function(θ_RVs, x_RVs + z_RVs)

but don’t know how to do the same all in terms of the transformed variables? Any tips?