mamba-rs 0.2.1

Mamba SSM (Selective State Space Model) in Rust with CUDA GPU support. Training + inference, forward + backward (BPTT), burn-in, custom CUDA kernels.
Documentation 
1
2
3
4
5
6
7
8

//! High-level Mamba wrappers.
//!
//! [`MambaBackbone`] is the primary user-facing API. It owns all weights
//! and provides both single-step inference and batched training methods.

mod backbone;

pub use backbone::MambaBackbone;