probml / dynamax

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

add support for storing just means or covs during kalman filtering #291

Open murphyk opened 1 year ago

murphyk commented 1 year ago

Following https://github.com/probml/dynamax/pull/289, add similar functionality to the linear Gaussian code.