NeMo-AutoModel
Table of Contents
🖥️ Environment Start
🚀 E2E Examples
📚 Guides
🛠️ Development
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.vlm
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.stateful_wrappers
nemo_automodel.checkpoint.checkpointing
nemo_automodel.checkpoint._torch_backports
