API Reference#
This page contains auto-generated API reference documentation [1].
nemo_automodel
nemo_automodel.components
nemo_automodel.components._diffusers
nemo_automodel.components._peft
nemo_automodel.components._transformers
nemo_automodel.components.attention
nemo_automodel.components.checkpoint
nemo_automodel.components.checkpoint._backports
nemo_automodel.components.checkpoint._backports._fsspec_filesystem
nemo_automodel.components.checkpoint._backports._version
nemo_automodel.components.checkpoint._backports.consolidate_hf_safetensors
nemo_automodel.components.checkpoint._backports.default_planner
nemo_automodel.components.checkpoint._backports.filesystem
nemo_automodel.components.checkpoint._backports.hf_storage
nemo_automodel.components.checkpoint._backports.hf_utils
nemo_automodel.components.checkpoint._backports.planner_helpers
nemo_automodel.components.checkpoint._torch_backports
nemo_automodel.components.checkpoint.checkpointing
nemo_automodel.components.checkpoint.state_dict_adapter
nemo_automodel.components.checkpoint.stateful_wrappers
nemo_automodel.components.config
nemo_automodel.components.datasets
nemo_automodel.components.datasets.llm
nemo_automodel.components.datasets.llm.megatron
nemo_automodel.components.datasets.llm.megatron.builder
nemo_automodel.components.datasets.llm.megatron.gpt_dataset
nemo_automodel.components.datasets.llm.megatron.helpers
nemo_automodel.components.datasets.llm.megatron.indexed_dataset
nemo_automodel.components.datasets.llm.megatron.megatron_utils
nemo_automodel.components.datasets.llm.megatron.sampler
nemo_automodel.components.datasets.llm.column_mapped_text_instruction_dataset
nemo_automodel.components.datasets.llm.formatting_utils
nemo_automodel.components.datasets.llm.hellaswag
nemo_automodel.components.datasets.llm.megatron_dataset
nemo_automodel.components.datasets.llm.mock
nemo_automodel.components.datasets.llm.mock_iterable_dataset
nemo_automodel.components.datasets.llm.mock_packed
nemo_automodel.components.datasets.llm.nanogpt_dataset
nemo_automodel.components.datasets.llm.packed_sequence
nemo_automodel.components.datasets.llm.squad
nemo_automodel.components.datasets.llm.tool_calling_chat_dataset
nemo_automodel.components.datasets.vlm
nemo_automodel.components.datasets.utils
nemo_automodel.components.distributed
nemo_automodel.components.distributed.pipelining
nemo_automodel.components.distributed.cp_utils
nemo_automodel.components.distributed.ddp
nemo_automodel.components.distributed.fsdp2
nemo_automodel.components.distributed.grad_utils
nemo_automodel.components.distributed.init_utils
nemo_automodel.components.distributed.megatron_fsdp
nemo_automodel.components.distributed.optimized_tp_plans
nemo_automodel.components.distributed.parallel_styles
nemo_automodel.components.distributed.parallelizer
nemo_automodel.components.distributed.tensor_utils
nemo_automodel.components.distributed.utils
nemo_automodel.components.launcher
nemo_automodel.components.loggers
nemo_automodel.components.loss
nemo_automodel.components.models
nemo_automodel.components.moe
nemo_automodel.components.moe.megatron
nemo_automodel.components.moe.fsdp_mixin
nemo_automodel.components.moe.layers
nemo_automodel.components.moe.parallelizer
nemo_automodel.components.moe.rope_utils
nemo_automodel.components.moe.state_dict_mixin
nemo_automodel.components.moe.state_dict_utils
nemo_automodel.components.moe.utils
nemo_automodel.components.optim
nemo_automodel.components.quantization
nemo_automodel.components.training
nemo_automodel.components.utils
nemo_automodel.shared