add computing the raw posterior embedding