If there is only one observation, it's possible that one of the transition kernels returns a weighted prob of Inf, leading to a error in `categorical` Also, using `categorical` here can be slow.
If there is only one observation, it's possible that one of the transition kernels returns a weighted prob of Inf, leading to a error in
categoricalAlso, using
categoricalhere can be slow.