For these next couple of weeks we’ll be reading through the non Linear Gaussian SSM notebooks in the dynamax docs, as well as the associated chapters in the ProbML book.
I have not identified those sections yet but when I do I’ll update this post!
https://probml.github.io/dynamax/