Enum MambaLatentNetConfig 

pub enum MambaLatentNetConfig {
    Mamba1 {
        input_size: usize,
        n_real_layers: usize,
        n_virtual_layers: Option<(usize, Schedule)>,
        mamba_block: Mamba1Config,
        output_size: usize,
        class_tokens: Vec<ClassToken>,
        ignore_first_residual: bool,
        ignore_last_residual: bool,
        residuals: ResidualsConfig,
    },
    Mamba2 {
        input_size: usize,
        n_real_layers: usize,
        n_virtual_layers: Option<(usize, Schedule)>,
        mamba_block: Mamba2Config,
        output_size: usize,
        class_tokens: Vec<ClassToken>,
        ignore_first_residual: bool,
        ignore_last_residual: bool,
        residuals: ResidualsConfig,
    },
    Mamba3 {
        input_size: usize,
        n_real_layers: usize,
        n_virtual_layers: Option<(usize, Schedule)>,
        mamba_block: Mamba3Config,
        output_size: usize,
        class_tokens: Vec<ClassToken>,
        ignore_first_residual: bool,
        ignore_last_residual: bool,
        residuals: ResidualsConfig,
    },
}

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

Variants

Mamba1

Build a Mamba-1 latent network.

Fields

input_size: usize

Input feature width.

n_real_layers: usize

Number of real layers.

n_virtual_layers: Option<(usize, Schedule)>

Optional virtual-layer scheduling.

mamba_block: Mamba1Config

Shared block config.

output_size: usize

Output feature width.

class_tokens: Vec<ClassToken>

Network-level class tokens, spliced into the input before in_proj.

ignore_first_residual: bool

Suppress the first virtual layer’s residual (Pre-LN skip / MultiGate seed carry). See Layers.

ignore_last_residual: bool

Suppress the last virtual layer’s residual (output is the last layer’s transform alone). See Layers.

residuals: ResidualsConfig

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

Mamba2

Build a Mamba-2 latent network.

Fields

input_size: usize

Input feature width.

n_real_layers: usize

Number of real layers.

n_virtual_layers: Option<(usize, Schedule)>

Optional virtual-layer scheduling.

mamba_block: Mamba2Config

Shared block config.

output_size: usize

Output feature width.

class_tokens: Vec<ClassToken>

Network-level class tokens, spliced into the input before in_proj.

ignore_first_residual: bool

Suppress the first virtual layer’s residual (Pre-LN skip / MultiGate seed carry). See Layers.

ignore_last_residual: bool

Suppress the last virtual layer’s residual (output is the last layer’s transform alone). See Layers.

residuals: ResidualsConfig

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

Mamba3

Build a Mamba-3 latent network.

Fields

input_size: usize

Input feature width.

n_real_layers: usize

Number of real layers.

n_virtual_layers: Option<(usize, Schedule)>

Optional virtual-layer scheduling.

mamba_block: Mamba3Config

Shared block config.

output_size: usize

Output feature width.

class_tokens: Vec<ClassToken>

Network-level class tokens, spliced into the input before in_proj.

ignore_first_residual: bool

Suppress the first virtual layer’s residual (Pre-LN skip / MultiGate seed carry). See Layers.

ignore_last_residual: bool

Suppress the last virtual layer’s residual (output is the last layer’s transform alone). See Layers.

residuals: ResidualsConfig

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

Implementations

impl MambaLatentNetConfig

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

Allocate and initialise the selected network on device.

Trait Implementations

impl Clone for MambaLatentNetConfig

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 MambaLatentNetConfig

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

Loads the configuration from a binary buffer.

impl Debug for MambaLatentNetConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for MambaLatentNetConfig

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

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for MambaLatentNetConfig

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

Formats the value using the given formatter.

impl Serialize for MambaLatentNetConfig

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

where S: Serializer,

Serialize this value into the given Serde serializer.