silvermpx / mamba-rsView on GitHub
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.
☆19Oct 20, 2026Updated this week

Alternatives and similar repositories for mamba-rs

Users that are interested in mamba-rs are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?