Using DiscreteMarkovChain with batch dimensions and data of different lengths

Thank you so much!