I think what I am doing is observed on the marginal, and also correct for the correlation using pm.Potential, you do need both of them to work, and the marginal does helps a lot of model convergence as you can see in the models with marginal only.
In your code you were not correcting for the Jacobian? you do need it for the estimation to be accurate.