Initialization problem sampling from discrete distribution

numpyro
2

Are you using enumeration? It seems that your model violates this assumption. It is better to use DiscreteHMCGibbs I think.