Thank you for your answer!
I cannot find in the documentation what that function does to my variables. Either way, I implemented it and the inferences were much worse.
I think that, in my case, the best way to handle this is by post processing the trace. I don’t have mid chain label switching, so I can just re-label my components.