for (g in 1:n_rates) {
    real chi_mu = mu_pop_level[1]*exp(
        (alpha_dev+sigma_dev*child_dev_speech_age[speech_rate_child[g]])*speech_rate_age[g]/12.0/10.0 + (speech_rate_child_level[speech_rate_child[g],2]+speech_rate_child_level[speech_rate_child[g],3]-mu_pop_level[3]-mu_pop_level[4])*(beta_dev*speech_rate_age[g]/12.0/10.0+beta_direct/10.0)
    );
    speech_rate[1,g] ~ gamma(
        alpha_child_level[1],
        alpha_child_level[1]/chi_mu
    );

    speech_rate[2:,g] ~ gamma(
        alpha_child_level[2:],
        (alpha_child_level[2:]./(speech_rate_child_level[speech_rate_child[g],:]')) //'
    );
    speech_rates[g,:] ~ poisson(speech_rate[:,g]*durations[g]*1000);
}

for (c in 1:n_speech_rate_children) {
    int distrib = child_siblings[c]==0?2:1;

    speech_rate_child_level[c,1] ~ gamma(
        alpha_corpus_level[distrib,1],
        (alpha_corpus_level[distrib,1]/(mu_corpus_level[1,speech_rate_child_corpus[c]]*exp(
            speech_rate_child_siblings[c]==0?beta_sib_och:0
        )))
    );

    speech_rate_child_level[c,2:] ~ gamma(
        alpha_corpus_level[distrib,2:],
        (alpha_corpus_level[distrib,2:]./(mu_corpus_level[2:,speech_rate_child_corpus[c]]*exp(
            speech_rate_child_siblings[c]==0?beta_sib_adu/10.0:0
        )))
    );
}

child_dev_speech_age ~ normal(0, 1);