probml / dynamax

State Space Models library in JAX
https://probml.github.io/dynamax/
MIT License
667 stars 76 forks source link

add block gibbs sampling for HMM learning #276

Open murphyk opened 1 year ago

murphyk commented 1 year ago

We currently support block gibbs sampling for LGSSM (see here). We should do the same for HMMs, assuming conjugate priors.