Skip to main content

Module network

Module network 

Source
Expand description

Family-generic networks (MambaLatentNet / MambaVocabNet).

Structs§

LatentNetwork
A feature/regression network on latents: in_proj (input_size → d_model) → Layers<M> → out_proj (d_model → output_size).
LatentNetworkBuilder
Plain factory for LatentNetwork.
LatentNetworkRecord
The record type for the module.
LatentNetworkRecordItem
The record item type for the module.
VocabNetwork
A complete autoregressive language model over a token vocabulary: Embedding (vocab → d_model) → Layers<M> → norm_f → LM head (d_model → vocab).
VocabNetworkBuilder
Plain factory for VocabNetwork. Mirrors LatentNetworkBuilder but adds vocab padding and the tied/untied LM-head choice.
VocabNetworkRecord
The record type for the module.
VocabNetworkRecordItem
The record item type for the module.

Enums§

MambaLatentNet
A runtime-selectable latent network: the same in_proj → Layers → out_proj shape over any Mamba-x family, chosen at runtime.
MambaLatentNetConfig
The serializable, documentation-friendly config for MambaLatentNet. Each variant is concrete (per-family), so #[derive(Config)] applies; init builds the matching network variant.
MambaLatentNetConfigSerde 🔒
MambaLatentNetRecord
The record type for the module.
MambaLatentNetRecordItem
The record item type for the module.
MambaVocabNet
A runtime-selectable token language model: the same Embedding → Layers → norm_f → LM head shape over any Mamba-x family, chosen at runtime. The vocabulary counterpart of MambaLatentNet.
MambaVocabNetConfig
The serializable, documentation-friendly config for MambaVocabNet. Each variant is concrete (per-family), so #[derive(Config)] applies; init builds the matching network variant.
MambaVocabNetConfigSerde 🔒
MambaVocabNetRecord
The record type for the module.
MambaVocabNetRecordItem
The record item type for the module.