How about the implementation of Mamba-3?

[Syaoran036]

https://openreview.net/pdf?id=HwCvaJOiCj

I'm really looking forward to this

[swfsql]

Not wanting to promote here, but I did give it a try with an AI in here - didn't test tho.

note: the forward is not using the semi-separable decomposition (ssd), just the same as the step; this is how ssd looks like for Mamba-2, I think.