Hidden semi markov model using tfd.HiddenMarkovModel

[lucasmiranda42]

Dear maintainers,

I was wondering if there is any way of extending tfd.HiddenMarkovModel to a hidden semi-Markov setting, where emissions depend not only on the last seen hidden state but also on the time elapsed since the last state switch (see figure below).

Thanks a lot for your help! Lucas