Как реализовать LDA со свернутой выборкой Гиббса с помощью R-JAGS?

Question

Я пытаюсь реализовать модель LDA с JAGS:

model {

    for (i in 1:M) {
      theta[i,1:K] ~ ddirch(alpha)
    }


    for (i in 1:K) {
      phi[i,1:V] ~ ddirch(beta)
    }


    for (i in 1:W) {
      z[i] ~ dcat(theta[D[i],])
      w[i] ~ dcat(phi[z[i],])
    }
   }

Программа потерпела крах, поскольку выборка Гиббса для LDA слишком тяжелая. Теперь я хотел бы реализовать это с collapsed Gibbs sampling, Вы знаете, как пробовать таким образом с JAGS? Заранее спасибо!