How to implement the NUTS algorithm in numpyro when (no downstream coupling) is violated?

I notice that when restriction (no downstream coupling)http://pyro.ai/examples/enumeration.html is violated, the sequential plate should be used. However, the error (This algorithm might only work for discrete sites with enumerate marked 'parallel')appears when I simultaneous setting infer={"enumerate": "sequential"} and using the NUTS algorithm.
My model is very similar to Unable implement a Cognitive Model - DINA in Pyro.
I also tried both mix HMC and DiscreteHMCGibbs, but mix HMC gets stuck when the sample size is large, and DiscreteHMCGibbs runs very slowly.So, how to implement the NUTS algorithm when (no downstream coupling) is violated?
Thanks a lot!