Struct SSMConfig 

pub struct SSMConfig {
    pub d_model: usize,
    pub d_state: usize,
    pub d_conv: usize,
    pub expand_factor: usize,
    pub dt_rank: usize,
}

Configuration for a Selective State Space Model layer.

Fields

d_model: usize

Model dimension (input/output width).

d_state: usize

State dimension (N). Controls the capacity of the recurrent state.

d_conv: usize

1D convolution kernel size. Provides local context before the SSM.

expand_factor: usize

Inner dimension expansion factor. The SSM operates at d_model * expand.

dt_rank: usize

Rank of the Delta projection (dt_rank). Lower rank saves parameters.

Implementations

impl SSMConfig

pub fn new(d_model: usize) -> Self

Creates a config with sensible defaults matching Mamba-130M.

pub fn d_inner(&self) -> usize

The inner (expanded) dimension used inside the SSM block.

pub fn validate(&self) -> Result<(), &'static str>

Validates the configuration, returning an error message if invalid.

Trait Implementations

impl Clone for SSMConfig

fn clone(&self) -> SSMConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for SSMConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.