GitHub repository: vvvm23/mamba-jax
Primary language: Python
Description: Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX