Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/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.
efficient mamba pytorch selective-scan
Last synced: 31 Jul 2024