mamba-rs
A complete Mamba implementation in Rust: Mamba, Mamba-2 and Mamba-3 (SISO and MIMO), inference and training on the CPU and on NVIDIA GPUs with the crate's own NVRTC-compiled kernels, f32 / tf32 / bf16 / f16, CUDA graphs, official checkpoints, deterministic GEMMs by default with cuBLAS Fast and Pedantic modes.
Activity
- Latest release
- 22h ago
- Total releases
- 35
- Cadence
- ~daily
- Last 12 months
- 35
Reach
- Downloads
- 785
- Stars
- 20
Details
- License
- MIT OR Apache-2.0
- First release
- Mar 23, 2026
Releases