You may also be interested in https://github.com/rbitr/llm.f90/tree/master/ssm https://github.com/rbitr/llm.f90/tree/master/ssm it's my inference only implement

You may also be interested in https://github.com/rbitr/llm.f90/tree/master/ssm https://github.com/rbitr/llm.f90/tree/master/ssm it's my inference only implementation of mamba which ends up being much simpler than the training code in the original repo