Thanks everyone. @junpenglao That’s great that you can just pass it as a kwarg, terrific. I’ve marked that as the solution, thanks a lot!
I’m interested in modelling something that might have a lot of density around zero, so I thought of the square root as a possible option. But I do also see the possible problems of it not being bijective. I’ll have to think a bit!