NUTS pauses after init

I have also ran into similar situations, I assumed that the sampler is exploring in a region with points that cannot be accepted according to the acceptance criteria. Therefore, it will be seems as stuck until it find a sample that is acceptable.

I also notice that you are trying to draw 200000 samples, that may be the issue, try reducing that to 20000.