PrototypeCommon
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.
- Owner
- MzeroMiko (personal account)
- Stars
- Stars: 108
- Forks
- Forks: 3
- Language
- Python
- License
- Apache-2.0
- Created
- 2024
- Topics
- efficientmambapytorchselective-scan
- Tier
- Prototype (100 to 249 ★) What this tier means
More repos by MzeroMiko
All 3 repos by MzeroMiko- MzeroMiko/VMambaStars: 3,243PythonRC
- MzeroMiko/vHeatStars: 286PythonAlpha
Nearby in Python
All Python repos- yqf66/SoleviaStars: 108PythonPrototype
- BEN020406/maimen-order-managerStars: 108PythonPrototype
- Andreas698/glm-5.3-uncensoredStars: 108PythonPrototype
- drf/amsn2Stars: 108PythonPrototype
- dustin/twitty-twisterStars: 108PythonPrototype
- ept/uploadr.pyStars: 108PythonPrototype