Enum MambaBidiLayersConfig 

pub enum MambaBidiLayersConfig {
    Mamba1 {
        n_real_layers: usize,
        n_virtual_layers: Option<(usize, BidiSchedule)>,
        mamba_block: Mamba1Config,
        ignore_first_residual: bool,
        ignore_last_residual: bool,
        outputs_merge: Vec<OutputMergeConfig>,
        class_latents: Vec<ClassLatent>,
        residuals: ResidualsConfig,
    },
    Mamba2 {
        n_real_layers: usize,
        n_virtual_layers: Option<(usize, BidiSchedule)>,
        mamba_block: Mamba2Config,
        ignore_first_residual: bool,
        ignore_last_residual: bool,
        outputs_merge: Vec<OutputMergeConfig>,
        class_latents: Vec<ClassLatent>,
        residuals: ResidualsConfig,
    },
    Mamba3 {
        n_real_layers: usize,
        n_virtual_layers: Option<(usize, BidiSchedule)>,
        mamba_block: Mamba3Config,
        ignore_first_residual: bool,
        ignore_last_residual: bool,
        outputs_merge: Vec<OutputMergeConfig>,
        class_latents: Vec<ClassLatent>,
        residuals: ResidualsConfig,
    },
}

The serializable config for MambaBidiLayers. Each variant is concrete (per-family), so #[derive(Config)] applies; init builds the matching stack variant.

Variants

Mamba1

Build a Mamba-1 bidirectional stack.

Fields

n_real_layers: usize

Number of real layers (must be even — used in pairs).

n_virtual_layers: Option<(usize, BidiSchedule)>

mamba_block: Mamba1Config

Shared block config.

ignore_first_residual: bool

ignore_last_residual: bool

outputs_merge: Vec<OutputMergeConfig>

One merge config per pair, length n_real_layers / 2.

class_latents: Vec<ClassLatent>

Stack-level class latents, spliced into the sequence before the first pair (e.g. a Middle summary latent in place of mean-pooling).

residuals: ResidualsConfig

Inter-pair residual scheme (plain additive vs Multi-Gate).

Mamba2

Build a Mamba-2 bidirectional stack.

Fields

n_real_layers: usize

Number of real layers (must be even — used in pairs).

n_virtual_layers: Option<(usize, BidiSchedule)>

mamba_block: Mamba2Config

Shared block config.

ignore_first_residual: bool

ignore_last_residual: bool

outputs_merge: Vec<OutputMergeConfig>

One merge config per pair, length n_real_layers / 2.

class_latents: Vec<ClassLatent>

Stack-level class latents, spliced into the sequence before the first pair (e.g. a Middle summary latent in place of mean-pooling).

residuals: ResidualsConfig

Inter-pair residual scheme (plain additive vs Multi-Gate).

Mamba3

Build a Mamba-3 bidirectional stack.

Fields

n_real_layers: usize

Number of real layers (must be even — used in pairs).

n_virtual_layers: Option<(usize, BidiSchedule)>

mamba_block: Mamba3Config

Shared block config.

ignore_first_residual: bool

ignore_last_residual: bool

outputs_merge: Vec<OutputMergeConfig>

One merge config per pair, length n_real_layers / 2.

class_latents: Vec<ClassLatent>

Stack-level class latents, spliced into the sequence before the first pair (e.g. a Middle summary latent in place of mean-pooling).

residuals: ResidualsConfig

Inter-pair residual scheme (plain additive vs Multi-Gate).

Implementations

impl MambaBidiLayersConfig

pub fn init(&self, device: &Device) -> MambaBidiLayers

Allocate and initialise the selected bidirectional stack on device.

Trait Implementations

impl Clone for MambaBidiLayersConfig

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Config for MambaBidiLayersConfig

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for MambaBidiLayersConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for MambaBidiLayersConfig

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for MambaBidiLayersConfig

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

Formats the value using the given formatter.

impl Serialize for MambaBidiLayersConfig

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.