nemo_automodel

Subpackages

• nemo_automodel._transformers
  • nemo_automodel._transformers.auto_model
• nemo_automodel.shared
  • nemo_automodel.shared.import_utils
• nemo_automodel.training
  • nemo_automodel.training.utils
  • nemo_automodel.training.rng
  • nemo_automodel.training.base_recipe
  • nemo_automodel.training.step_scheduler
  • nemo_automodel.training.timers
• nemo_automodel.utils
  • nemo_automodel.utils.sig_utils
  • nemo_automodel.utils.model_utils
  • nemo_automodel.utils.dist_utils
  • nemo_automodel.utils.yaml_utils
• nemo_automodel.datasets
  • nemo_automodel.datasets.llm
    • nemo_automodel.datasets.llm.packed_sequence
    • nemo_automodel.datasets.llm.mock
    • nemo_automodel.datasets.llm.squad
    • nemo_automodel.datasets.llm.mock_packed
    • nemo_automodel.datasets.llm.hellaswag
  • nemo_automodel.datasets.vlm
    • nemo_automodel.datasets.vlm.collate_fns
    • nemo_automodel.datasets.vlm.utils
    • nemo_automodel.datasets.vlm.datasets
  • nemo_automodel.datasets.utils
• nemo_automodel.optim
  • nemo_automodel.optim.scheduler
• nemo_automodel.distributed
  • nemo_automodel.distributed.fsdp2
  • nemo_automodel.distributed.parallelizer
  • nemo_automodel.distributed.ddp
  • nemo_automodel.distributed.cp_utils
  • nemo_automodel.distributed.nvfsdp
  • nemo_automodel.distributed.init_utils
• nemo_automodel.config
  • nemo_automodel.config.loader
  • nemo_automodel.config.cli
• nemo_automodel.loss
  • nemo_automodel.loss.linear_ce
  • nemo_automodel.loss.chunked_ce
  • nemo_automodel.loss.masked_ce
• nemo_automodel.loggers
  • nemo_automodel.loggers.wandb_utils
  • nemo_automodel.loggers.log_utils
• nemo_automodel.checkpoint
  • nemo_automodel.checkpoint._backports
    • nemo_automodel.checkpoint._backports._fsspec_filesystem
    • nemo_automodel.checkpoint._backports.hf_storage
    • nemo_automodel.checkpoint._backports._version
    • nemo_automodel.checkpoint._backports.default_planner
    • nemo_automodel.checkpoint._backports.consolidate_hf_safetensors
    • nemo_automodel.checkpoint._backports.hf_utils
    • nemo_automodel.checkpoint._backports.planner_helpers
    • nemo_automodel.checkpoint._backports.filesystem
  • nemo_automodel.checkpoint.stateful_wrappers
  • nemo_automodel.checkpoint.checkpointing
  • nemo_automodel.checkpoint._torch_backports

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__

[‘_peft’, ‘config’, ‘datasets’, ‘distributed’, ‘loggers’, ‘loss’, ‘optim’, ‘training’, ‘_transformer…

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.