Thanks for the response! I had not tried this and it seems to work (with the small change of using self.approx.logp_norm / self.approx.logq_norm / self.approx.symbolic_randoms)!
Much appreciated! I still don’t quite understand why this works, but that is certainly not immediately problematic.
Many thanks again!