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