mamba.np is a pure NumPy implementation of Mamba.
Please refer to Install.md guide
$ 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