Implementation of Mamba-minimal in JAX
- Installation instructions
- Custom Pallas kernels
Mamba: Linear-Time Sequence Modeling with Selective State Spaces
@article{mamba,
title={Mamba: Linear-Time Sequence Modeling with Selective State Spaces},
author={Gu, Albert and Dao, Tri},
journal={arXiv preprint arXiv:2312.00752},
year={2023}
}