Hidden Markov model in pyro and numpyro I took this great unsupervised example written in pyro and translated it into numypro (jax).