burn_mamba/mamba1/mod.rs
1//! # Mamba-1
2//!
3//! The original selective state space model. Mamba-1 runs a **sequential
4//! selective scan** (no SSD, no backend-extension trait). See [`mamba1`] for
5//! the block; the residual layer stack, the full language model, and the
6//! bidirectional wrappers are the family-generic types in [`crate::generic`]
7//! (e.g. `MambaLatentNet` / `MambaVocabNet` / `MambaBidiLayers`).
8//!
9//! - [`mamba1`] — the selective-SSM block.
10//! - [`cache`] — the conv-window + SSM-state carried between calls.
11
12pub mod cache;
13pub mod mamba1;
14
15/// Public re-exports for Mamba-1.
16pub mod prelude {
17 use super::*;
18 pub use cache::{Mamba1Cache, Mamba1CacheConfig, Mamba1Caches, Mamba1CachesConfig};
19 pub use mamba1::{Mamba1, Mamba1Config};
20}