You’ll want to focus your attention on the sigma_theta trace — everything stalls when
sigma_theta hits zero. What’s happening here is that HMC (and just about everything)
is not geometrically ergodic for your model, and getting stuck like this is a manifestation
of the lack of geometric ergodicity. You’ll also note that n_divergent becomes non-zero
in this region as well, which is an HMC specific manifestation of the problem.
This is particularly important as geometric ergodicity is what guarantees that Monte Carlo
estimates converge to the true expectations (via the Monte Carlo Central Limit Theorem),
and in a case like this you have to be very dubious of the estimates. Note that algorithms
like Random Walk Metropolis and the Gibbs sampler suffer from the same problem, only
they’re usually not fast enough to be run for enough iterations to see the pathology manifest.