# On the Recall Scaling Laws in Mamba

This repo contains the code for ICLR 2026 submitted paper "On the Recall Scaling Laws in Mamba".

Full code will be released after peer review.

### Notebooks and Figures

Jupyter notebooks used to generate paper figures can be found in `scripts`.


### How to run?

Modify the configuration files in `config` as desired. Then: 

> docker build -t mamba-ar-docker .

> docker run --gpus all -it -v "$PWD":/workspace --entrypoint bash mamba-ar-docker


### Used repositories

This repo builds upon `mamba-ssm`, `mamba-tiny` and `zoology` open-source code repositories. 
We thank them for their important and helpful work.