Thanks @OriolAbril for the additional background. What exactly do you mean by “the same shape as the posterior”? Does it mean shape=(chain, draw) or shape=(chain, draw, #of RVs in the model) and that’s why it’s effectively doubling the memory footprint?