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.
Python
Stargazers
- AZZMM
- BackloryUniversity of Chinese Academy of Sciences
- BezdarnostKyrgyzstan
- cheng221
- ChiShengChen
- cuiwenyaonull
- dongzhuoyaoWhu->PKU->University of Amsterdam-> LMU
- haoranDUK
- Hiccupwzy
- huyhoang17HUST
- IanGuangleiZhu
- JeffCarpenterCanada
- johnnynunezBarcelona
- kuangliuZhejiang University
- kwonyoung9120Seoul National University
- lccworld
- mancevd
- MartinserNankai University
- minhtcaiCarnegie Mellon University
- MzeroMikoUCAS
- ncTimTang
- pmusacchioSeoul National University
- radarFudanNUS
- theAdamColton
- tic-top
- toqitahamidDhaka, Bangladesh
- usernameAjisai
- VToucher
- WingDust
- YouHuang67
- yuchongy1University of Melbourne
- YuheDAHU
- zhangkejiang
- zmxxmz0713
- zwy0913
- zxd3099Wuhan University