API Reference

This page contains auto-generated API reference documentation [1].

• nemo_automodel
  • 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

[1]

Created with sphinx-autodoc2