Sign in Sign up
cargo

mamba-rs

Mamba SSM and Mamba-3 SISO in Rust with optional CUDA acceleration: inference and training (BPTT through the SSM state, AdamW) on CPU and GPU, custom NVRTC-compiled kernels, CUDA Graph capture, f32 / bf16 / f16 storage. Deterministic batch-invariant GEMMs by default, bit-identical run to run and across batch sizes; cuBLAS Fast and Pedantic by name.

cargo View on Cargo github View on Github

Activity

Latest release
2d ago
Total releases
27
Cadence
~daily
Last 12 months
27

Reach

Downloads
680
Stars
17

Details

License
MIT OR Apache-2.0
First release
Mar 23, 2026
Releases
Version Released
0.7.0 minor
0.6.9 patch
0.6.8 patch
0.6.7 patch
0.6.6 patch
0.6.5 patch
0.6.4 patch
0.6.3 patch
0.6.2 patch
0.6.1 patch
0.6.0 minor
0.5.3 patch
0.5.2 patch
0.5.1 patch
0.5.0 minor
0.4.2 patch
0.4.1 patch
0.4.0 minor
0.3.1 patch
0.3.0 minor
0.2.1 patch
0.2.0 minor
0.1.4 patch
0.1.3 patch
0.1.2 patch
0.1.1 patch
0.1.0 initial