Thanks Ricardo! The reason is that I think a particular variational approximation might be more reasonable for the square root transformation than for the log transformation. There’s a discussion on Section 3.3 here for example of why different transformations might be worth exploring: https://arxiv.org/pdf/1603.00788.pdf