Expand description
Bidirectional layer stacks (straight + reversed passes, merged per pair).
Structs§
- Bidi
Layer Pair - A single bidirectional pair: a straight (→) and a reversed (←) Pre-LN block
whose outputs are merged. The residual is not applied here — the
enclosing
BidiLayersadds it (or suppresses it on the first/last pair), mirroring theLayer/Layerssplit. - Bidi
Layer Pair Record - The record type for the module.
- Bidi
Layer Pair Record Item - The record item type for the module.
- Bidi
Layers - A stack of bidirectional
Layerpairs with optional virtual-layer scheduling — one struct for every Mamba-x family. - Bidi
Layers Builder - Plain (non-serde) factory for
BidiLayers. - Bidi
Layers Record - The record type for the module.
- Bidi
Layers Record Item - The record item type for the module.
- NoOp
- A zero-parameter placeholder for the parameterless
Meanmerge.
Enums§
- Mamba
Bidi Layers - A runtime-selectable bidirectional stack: the same paired straight/reverse
structure over any Mamba-x family, chosen at runtime. The forward-only
counterpart of
MambaLatentNetfor non-autoregressive tasks. - Mamba
Bidi Layers Config - The serializable config for
MambaBidiLayers. Each variant is concrete (per-family), so#[derive(Config)]applies;initbuilds the matching stack variant. - Mamba
Bidi 🔒Layers Config Serde - Mamba
Bidi Layers Record - The record type for the module.
- Mamba
Bidi Layers Record Item - The record item type for the module.
- Output
Merge - How the two directions of a bidirectional pair are combined.
- Output
Merge Config - Configuration / factory for
OutputMerge. - Output
Merge 🔒Config Serde - Output
Merge Record - The record type for the module.
- Output
Merge Record Item - The record item type for the module.
Functions§
- bidi_
pair_ 🔒forward - The straight + reverse + merge computation of a bidirectional pair, over borrowed sub-modules.