Solved by googlebot in post #3 
              
              
                check out scale_tril argument to MultivariateNormal (as docs say, this is the primary parameterization), or LowRankMultivariateNormal