nemo_automodel

Subpackages

• nemo_automodel._diffusers
  • nemo_automodel._diffusers.auto_diffusion_pipeline
• nemo_automodel._transformers
  • nemo_automodel._transformers.auto_model
  • nemo_automodel._transformers.auto_tokenizer
  • nemo_automodel._transformers.registry
  • nemo_automodel._transformers.utils
• nemo_automodel.components
  • nemo_automodel.components._peft
    • nemo_automodel.components._peft.lora
    • nemo_automodel.components._peft.lora_kernel
    • nemo_automodel.components._peft.module_matcher
  • nemo_automodel.components.attention
    • nemo_automodel.components.attention.flex_attention
    • nemo_automodel.components.attention.utils
  • nemo_automodel.components.checkpoint
    • nemo_automodel.components.checkpoint._backports
    • nemo_automodel.components.checkpoint._torch_backports
    • nemo_automodel.components.checkpoint.addons
    • nemo_automodel.components.checkpoint.checkpointing
    • nemo_automodel.components.checkpoint.state_dict_adapter
    • nemo_automodel.components.checkpoint.stateful_wrappers
  • nemo_automodel.components.config
    • nemo_automodel.components.config._arg_parser
    • nemo_automodel.components.config.loader
  • nemo_automodel.components.datasets
    • nemo_automodel.components.datasets.llm
    • nemo_automodel.components.datasets.vlm
    • nemo_automodel.components.datasets.utils
  • nemo_automodel.components.distributed
    • nemo_automodel.components.distributed.pipelining
    • nemo_automodel.components.distributed.cp_utils
    • nemo_automodel.components.distributed.ddp
    • nemo_automodel.components.distributed.fsdp2
    • nemo_automodel.components.distributed.grad_utils
    • nemo_automodel.components.distributed.init_utils
    • nemo_automodel.components.distributed.megatron_fsdp
    • nemo_automodel.components.distributed.optimized_tp_plans
    • nemo_automodel.components.distributed.parallel_styles
    • nemo_automodel.components.distributed.parallelizer
    • nemo_automodel.components.distributed.parallelizer_utils
    • nemo_automodel.components.distributed.tensor_utils
    • nemo_automodel.components.distributed.thd_utils
    • nemo_automodel.components.distributed.utils
  • nemo_automodel.components.launcher
  • nemo_automodel.components.loggers
    • nemo_automodel.components.loggers.log_utils
    • nemo_automodel.components.loggers.metric_logger
    • nemo_automodel.components.loggers.mlflow_utils
    • nemo_automodel.components.loggers.wandb_utils
  • nemo_automodel.components.loss
    • nemo_automodel.components.loss.chunked_ce
    • nemo_automodel.components.loss.kd_loss
    • nemo_automodel.components.loss.linear_ce
    • nemo_automodel.components.loss.masked_ce
    • nemo_automodel.components.loss.te_parallel_ce
  • nemo_automodel.components.models
    • nemo_automodel.components.models.biencoder
    • nemo_automodel.components.models.common
    • nemo_automodel.components.models.deepseek_v3
    • nemo_automodel.components.models.glm4_moe
    • nemo_automodel.components.models.gpt_oss
    • nemo_automodel.components.models.llama
    • nemo_automodel.components.models.qwen2
    • nemo_automodel.components.models.qwen3_moe
    • nemo_automodel.components.models.qwen3_next
    • nemo_automodel.components.models.qwen3_omni_moe
    • nemo_automodel.components.models.qwen3_vl_moe
    • nemo_automodel.components.models.gpt2
  • nemo_automodel.components.moe
    • nemo_automodel.components.moe.megatron
    • nemo_automodel.components.moe.fsdp_mixin
    • nemo_automodel.components.moe.layers
    • nemo_automodel.components.moe.parallelizer
    • nemo_automodel.components.moe.state_dict_mixin
    • nemo_automodel.components.moe.state_dict_utils
    • nemo_automodel.components.moe.utils
  • nemo_automodel.components.optim
    • nemo_automodel.components.optim.scheduler
  • nemo_automodel.components.quantization
    • nemo_automodel.components.quantization.fp8
    • nemo_automodel.components.quantization.qat
    • nemo_automodel.components.quantization.qlora
  • nemo_automodel.components.training
    • nemo_automodel.components.training.rng
    • nemo_automodel.components.training.signal_handler
    • nemo_automodel.components.training.step_scheduler
    • nemo_automodel.components.training.timers
    • nemo_automodel.components.training.utils
  • nemo_automodel.components.utils
    • nemo_automodel.components.utils.compile_utils
    • nemo_automodel.components.utils.flops_utils
    • nemo_automodel.components.utils.model_utils
    • nemo_automodel.components.utils.yaml_utils
• nemo_automodel.shared
  • nemo_automodel.shared.import_utils
  • nemo_automodel.shared.utils

Package Contents

Functions

__getattr__	Lazily import and cache submodules listed in all when accessed.
__dir__	Expose the names of all available submodules for auto-completion.

Data

__all__

API

nemo_automodel.__all__

[‘recipes’, ‘shared’, ‘components’, ‘version’, ‘package_name’]

nemo_automodel.__getattr__(name: str)

Lazily import and cache submodules listed in all when accessed.

Raises:

AttributeError if the name isn’t in all. –

nemo_automodel.__dir__()

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