Yeah, it is really that each element of theta_raw has the same unit-scale (a priori) and is independent of mu, sigma, and everything else. If you made theta primitive and did theta[k] ~ normal(mu, sigma) none of that holds, which usually makes it more difficult for NUTS.