nemo_automodel.components.moe

This page is for version 0.5.0 · 26.06. For other versions, use one of these documentation indexes: - Nightly (default): https://docs.nvidia.com/nemo/automodel/nightly/llms.txt - Latest: https://docs.nvidia.com/nemo/automodel/latest/llms.txt - 0.5.0 · 26.06: https://docs.nvidia.com/nemo/automodel/v0.5/llms.txt - 0.4.0 · 26.04: https://docs.nvidia.com/nemo/automodel/v0.4/llms.txt

Subpackages

• nemo_automodel.components.moe.megatron
• nemo_automodel.components.moe.uccl_ep

Submodules

• nemo_automodel.components.moe.config
• nemo_automodel.components.moe.experts
• nemo_automodel.components.moe.fsdp_mixin
• nemo_automodel.components.moe.layers
• nemo_automodel.components.moe.load_balance_metrics
• nemo_automodel.components.moe.mok_experts
• nemo_automodel.components.moe.mxfp8
• nemo_automodel.components.moe.optimized_ops
• nemo_automodel.components.moe.parallelizer
• nemo_automodel.components.moe.quantized_experts
• nemo_automodel.components.moe.router_replay
• nemo_automodel.components.moe.state_dict_mixin
• nemo_automodel.components.moe.state_dict_utils
• nemo_automodel.components.moe.swiglu_triton
• nemo_automodel.components.moe.tp_plan_validation