`core.models.mamba.mamba_layer_specs`#

Module Contents#

Data#

`moe`
`moe_inference`
`_mamba_mtp_block_spec`
`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.moe_inference#: ‘get_inference_optimized_moe_spec(…)’

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

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

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