All modules for which code is available

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