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.datasets
    • nemo_automodel.components.datasets.vlm
    • nemo_automodel.components.datasets.llm
    • 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.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.cp_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.parallel_styles
    • nemo_automodel.components.distributed.init_utils
    • nemo_automodel.components.distributed.fsdp2
    • nemo_automodel.components.distributed.optimized_tp_plans
    • 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.sig_utils
  • nemo_automodel.components.moe
    • nemo_automodel.components.moe.megatron
    • nemo_automodel.components.moe.utils
    • nemo_automodel.components.moe.layers
    • nemo_automodel.components.moe.rope_utils
    • nemo_automodel.components.moe.state_dict_mixin
    • nemo_automodel.components.moe.state_dict_utils
  • nemo_automodel.components.loggers
    • nemo_automodel.components.loggers.log_utils
    • nemo_automodel.components.loggers.wandb_utils
  • nemo_automodel.components.optim
    • nemo_automodel.components.optim.scheduler
  • nemo_automodel.components._transformers
    • nemo_automodel.components._transformers.auto_model
    • nemo_automodel.components._transformers.utils
  • nemo_automodel.components.checkpoint
    • nemo_automodel.components.checkpoint._backports
    • nemo_automodel.components.checkpoint._torch_backports
    • nemo_automodel.components.checkpoint.stateful_wrappers
    • nemo_automodel.components.checkpoint.state_dict_adapter
    • nemo_automodel.components.checkpoint.checkpointing
  • nemo_automodel.components.launcher
  • nemo_automodel.components.quantization
    • nemo_automodel.components.quantization.qlora
    • nemo_automodel.components.quantization.fp8
  • nemo_automodel.components.models
    • nemo_automodel.components.models.deepseek_v3
    • nemo_automodel.components.models.gpt2
• nemo_automodel.shared
  • nemo_automodel.shared.utils
  • nemo_automodel.shared.import_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.