GitHub / 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.
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/MzeroMiko%2Fmamba-mini
PURL: pkg:github/MzeroMiko/mamba-mini
Stars: 80
Forks: 1
Open issues: 4
License: None
Language: Python
Size: 1.16 MB
Dependencies parsed at: Pending
Created at: over 1 year ago
Updated at: 5 months ago
Pushed at: over 1 year ago
Last synced at: 5 months ago
Topics: efficient, mamba, pytorch, selective-scan