MzeroMiko / mamba-mini

An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivation. It is probably the code which is the most close to selective_scan_cuda in mamba.
87Updated last year

Alternatives and similar repositories for mamba-mini

Users that are interested in mamba-mini are comparing it to the libraries listed below

Sorting: