https://akillness.github.io/posts/ssm-transformer/