refactor EM for LGSSM to absorb bias term into input vector

Closed this issue 2 years ago · 0 comments

murphyk commented 2 years ago

Refactor https://github.com/probml/ssm-jax/blob/main/ssm_jax/lgssm/models.py#L214 so it has 4 blocks instead of 9.