`core.models.mamba.mamba_layer_specs`#

Module Contents#

Data#

`moe`
`mamba_stack_spec`
`mamba_inference_stack_spec`

API#

core.models.mamba.mamba_layer_specs.moe#: ‘get_moe_module_spec(…)’

core.models.mamba.mamba_layer_specs.mamba_stack_spec#: ‘ModuleSpec(…)’

core.models.mamba.mamba_layer_specs.mamba_inference_stack_spec#: ‘ModuleSpec(…)’