probml / dynamax

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

Add general info form message passing algorithms for linear Gaussian SSMs #361

Open slinderman opened 2 months ago

slinderman commented 2 months ago

The current versions of information form message passing for LGSSMs use a hybrid representation of dynamics parameters and information form emission parameters. It is often helpful to have routines that operate directly on the block tridiagonal precision matrix and the precision-weighted mean. Examples of such implementations are:

(I've also written information form samplers and smoothers in JAX, along the lines of the last two links.)

We should add similar inference routines to the LGSSM info inference code.