silvermpx/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.

1 best points · 0 comments · 1 Hacker News post · 20 GitHub stars · View on GitHub

Hacker News discussions

• Show HN: Mamba SSM in Rust – training and inference with custom CUDA kernels (1 points, 0 comments, by silvermpx, 2026-03-23)