mamba.np is a pure NumPy implementation of Mamba.

$ python mamba.py "I have a dream that"
"""
I have a dream that I will be able to see the sunrise in the morning.

Token count: 18, elapsed: 18.57s, 1 tokens/s
"""

If you use or discuss mamba.np in your academic research, please cite the project to help spread awareness:

@misc{mamba.np,
  title = {mamba.np: pure NumPy implementation for Mamba},
  author = {Ido Hakimi}, 
  howpublished = {\url{https://github.com/idoh/mamba.np}},
  note = {mamba.np, MIT License}
  year = {2024},
}

Thank you to the creators of the following libraries and tools and their contributors:

• mamba-minimal - @johnma2006
• llama3.np - @likejazz
• The Mamba architecture was introduced in Mamba: Linear-Time Sequence Modeling with Selective State Spaces by Albert Gu and Tri Dao
• The official implementation is here: https://github.com/state-spaces/mamba
• Title image was generated by Microsoft Designer