nemo_automodel

Subpackages

• nemo_automodel.components
  • nemo_automodel.components.loss
    • nemo_automodel.components.loss.chunked_ce
    • nemo_automodel.components.loss.kd_loss
    • nemo_automodel.components.loss.masked_ce
    • nemo_automodel.components.loss.linear_ce
    • nemo_automodel.components.loss.te_parallel_ce
  • nemo_automodel.components.training
    • nemo_automodel.components.training.step_scheduler
    • nemo_automodel.components.training.utils
    • nemo_automodel.components.training.rng
    • nemo_automodel.components.training.timers
    • nemo_automodel.components.training.garbage_collection
    • nemo_automodel.components.training.signal_handler
    • nemo_automodel.components.training.model_output_utils
  • nemo_automodel.components.datasets
    • nemo_automodel.components.datasets.vlm
    • nemo_automodel.components.datasets.llm
    • nemo_automodel.components.datasets.reservoir_sampler
    • nemo_automodel.components.datasets.utils
  • nemo_automodel.components._peft
    • nemo_automodel.components._peft.lora
    • nemo_automodel.components._peft.module_matcher
    • nemo_automodel.components._peft.lora_kernel
    • nemo_automodel.components._peft.lora_moe
  • nemo_automodel.components.config
    • nemo_automodel.components.config._arg_parser
    • nemo_automodel.components.config.loader
  • nemo_automodel.components.distributed
    • nemo_automodel.components.distributed.pipelining
    • nemo_automodel.components.distributed.parallelizer_utils
    • nemo_automodel.components.distributed.cp_utils
    • nemo_automodel.components.distributed.thd_utils
    • nemo_automodel.components.distributed.grad_utils
    • nemo_automodel.components.distributed.utils
    • nemo_automodel.components.distributed.ddp
    • nemo_automodel.components.distributed.megatron_fsdp
    • nemo_automodel.components.distributed.parallelizer
    • nemo_automodel.components.distributed.config
    • nemo_automodel.components.distributed.parallel_styles
    • nemo_automodel.components.distributed.init_utils
    • nemo_automodel.components.distributed.mesh
    • nemo_automodel.components.distributed.fsdp2
    • nemo_automodel.components.distributed.optimized_tp_plans
    • nemo_automodel.components.distributed.mesh_utils
    • nemo_automodel.components.distributed.tensor_utils
  • nemo_automodel.components.utils
    • nemo_automodel.components.utils.compile_utils
    • nemo_automodel.components.utils.model_utils
    • nemo_automodel.components.utils.yaml_utils
    • nemo_automodel.components.utils.flops_utils
  • nemo_automodel.components.moe
    • nemo_automodel.components.moe.megatron
    • nemo_automodel.components.moe.fsdp_mixin
    • nemo_automodel.components.moe.parallelizer
    • nemo_automodel.components.moe.config
    • nemo_automodel.components.moe.layers
    • nemo_automodel.components.moe.state_dict_mixin
    • nemo_automodel.components.moe.state_dict_utils
    • nemo_automodel.components.moe.load_balance_metrics
    • nemo_automodel.components.moe.experts
  • nemo_automodel.components.loggers
    • nemo_automodel.components.loggers.mlflow_utils
    • nemo_automodel.components.loggers.metric_logger
    • nemo_automodel.components.loggers.log_utils
    • nemo_automodel.components.loggers.wandb_utils
  • nemo_automodel.components.optim
    • nemo_automodel.components.optim.scheduler
    • nemo_automodel.components.optim.utils
  • nemo_automodel.components.checkpoint
    • nemo_automodel.components.checkpoint._backports
    • nemo_automodel.components.checkpoint.conversion_mapping
    • nemo_automodel.components.checkpoint.utils
    • nemo_automodel.components.checkpoint._torch_backports
    • nemo_automodel.components.checkpoint.stateful_wrappers
    • nemo_automodel.components.checkpoint.addons
    • nemo_automodel.components.checkpoint.state_dict_adapter
    • nemo_automodel.components.checkpoint.checkpointing
  • nemo_automodel.components.launcher
  • nemo_automodel.components.attention
    • nemo_automodel.components.attention.utils
    • nemo_automodel.components.attention.flex_attention
  • nemo_automodel.components.quantization
    • nemo_automodel.components.quantization.qlora
    • nemo_automodel.components.quantization.fp8
    • nemo_automodel.components.quantization.qat
  • nemo_automodel.components.models
    • nemo_automodel.components.models.common
    • nemo_automodel.components.models.llama_bidirectional
    • nemo_automodel.components.models.nemotron_parse
    • nemo_automodel.components.models.qwen3_moe
    • nemo_automodel.components.models.nemotron_v3
    • nemo_automodel.components.models.minimax_m2
    • nemo_automodel.components.models.kimivl
    • nemo_automodel.components.models.mistral3
    • nemo_automodel.components.models.kimi_k25_vl
    • nemo_automodel.components.models.glm4_moe_lite
    • nemo_automodel.components.models.qwen2
    • nemo_automodel.components.models.qwen3_next
    • nemo_automodel.components.models.step3p5
    • nemo_automodel.components.models.glm4_moe
    • nemo_automodel.components.models.deepseek_v32
    • nemo_automodel.components.models.qwen3_vl_moe
    • nemo_automodel.components.models.gpt_oss
    • nemo_automodel.components.models.llama
    • nemo_automodel.components.models.deepseek_v3
    • nemo_automodel.components.models.qwen3_omni_moe
    • nemo_automodel.components.models.qwen3_5_moe
    • nemo_automodel.components.models.gpt2
• nemo_automodel._diffusers
  • nemo_automodel._diffusers.auto_diffusion_pipeline
• nemo_automodel.autonvtx
• nemo_automodel._transformers
  • nemo_automodel._transformers.auto_model
  • nemo_automodel._transformers.biencoder
  • nemo_automodel._transformers.utils
  • nemo_automodel._transformers.model_init
  • nemo_automodel._transformers.auto_tokenizer
  • nemo_automodel._transformers.infrastructure
  • nemo_automodel._transformers.registry
  • nemo_automodel._transformers.kernel_patches
• nemo_automodel.shared
  • nemo_automodel.shared.utils
  • nemo_automodel.shared.import_utils
  • nemo_automodel.shared.torch_patches

Package Contents

Functions

__getattr__	Lazily import and cache selected submodules / exported symbols when accessed.
__dir__	Expose the names of all available submodules for auto-completion.

Data

_SUBMODULES
_LAZY_ATTRS
__all__

API

nemo_automodel._SUBMODULES

None

nemo_automodel._LAZY_ATTRS: dict[str, tuple[str, str]]

None

nemo_automodel.__all__

‘sorted(…)’

nemo_automodel.__getattr__(name: str)

Lazily import and cache selected submodules / exported symbols when accessed.

Raises:

AttributeError if the name isn’t in all. –

nemo_automodel.__dir__()

Expose the names of all available submodules for auto-completion.