Request for Implementation of Mamba Paper

The title explains itself. Here are some relevant resources:

• Official Mamba repository: https://github.com/state-spaces/mamba/blob/main/mamba_ssm/modules/mamba_simple.py
• Original paper: https://arxiv.org/abs/2312.00752
• Additional resources:
  • A Visual Guide to Mamba and State Space Models
  • Mamba Notes

I would like to assist with this implementation in any way possible.