[junipertea]

It corresponds to the paper formulas as a courtesy. While descriptive comments or function signature might be nicer, it still is reasonably easy to follow along from the paper description! (Although, most of the complexity is hidden away in selective_scan_cuda.fwd())

[telotortium]

Implementation of selective_scan_cuda: https://github.com/state-spaces/mamba/blob/6dbfc4553a98c81e1...

[triyambakam]

Aha, thanks, that is helpful. Any chance you know of a guide that walks through a paper and code side by side?

[ttul]

Hailing Jeremy Howard and his team at Fast.ai to make it so!