API Reference#
This page contains auto-generated API reference documentation [1].
nemo_automodelnemo_automodel._diffusersnemo_automodel._transformersnemo_automodel.componentsnemo_automodel.components._peftnemo_automodel.components.attentionnemo_automodel.components.checkpointnemo_automodel.components.checkpoint._backportsnemo_automodel.components.checkpoint._backports._fsspec_filesystemnemo_automodel.components.checkpoint._backports._versionnemo_automodel.components.checkpoint._backports.consolidate_hf_safetensorsnemo_automodel.components.checkpoint._backports.default_plannernemo_automodel.components.checkpoint._backports.filesystemnemo_automodel.components.checkpoint._backports.hf_storagenemo_automodel.components.checkpoint._backports.hf_utilsnemo_automodel.components.checkpoint._backports.planner_helpers
nemo_automodel.components.checkpoint._torch_backportsnemo_automodel.components.checkpoint.addonsnemo_automodel.components.checkpoint.checkpointingnemo_automodel.components.checkpoint.state_dict_adapternemo_automodel.components.checkpoint.stateful_wrappers
nemo_automodel.components.confignemo_automodel.components.datasetsnemo_automodel.components.datasets.llmnemo_automodel.components.datasets.llm.megatronnemo_automodel.components.datasets.llm.megatron.buildernemo_automodel.components.datasets.llm.megatron.gpt_datasetnemo_automodel.components.datasets.llm.megatron.helpersnemo_automodel.components.datasets.llm.megatron.indexed_datasetnemo_automodel.components.datasets.llm.megatron.megatron_utilsnemo_automodel.components.datasets.llm.megatron.sampler
nemo_automodel.components.datasets.llm.chat_datasetnemo_automodel.components.datasets.llm.column_mapped_text_instruction_datasetnemo_automodel.components.datasets.llm.column_mapped_text_instruction_iterable_datasetnemo_automodel.components.datasets.llm.formatting_utilsnemo_automodel.components.datasets.llm.hellaswagnemo_automodel.components.datasets.llm.megatron_datasetnemo_automodel.components.datasets.llm.mocknemo_automodel.components.datasets.llm.mock_iterable_datasetnemo_automodel.components.datasets.llm.mock_packednemo_automodel.components.datasets.llm.nanogpt_datasetnemo_automodel.components.datasets.llm.packed_sequencenemo_automodel.components.datasets.llm.retrieval_collatornemo_automodel.components.datasets.llm.retrieval_datasetnemo_automodel.components.datasets.llm.seq_clsnemo_automodel.components.datasets.llm.squad
nemo_automodel.components.datasets.vlmnemo_automodel.components.datasets.utils
nemo_automodel.components.distributednemo_automodel.components.distributed.pipeliningnemo_automodel.components.distributed.cp_utilsnemo_automodel.components.distributed.ddpnemo_automodel.components.distributed.fsdp2nemo_automodel.components.distributed.grad_utilsnemo_automodel.components.distributed.init_utilsnemo_automodel.components.distributed.megatron_fsdpnemo_automodel.components.distributed.optimized_tp_plansnemo_automodel.components.distributed.parallel_stylesnemo_automodel.components.distributed.parallelizernemo_automodel.components.distributed.parallelizer_utilsnemo_automodel.components.distributed.tensor_utilsnemo_automodel.components.distributed.thd_utilsnemo_automodel.components.distributed.utils
nemo_automodel.components.launchernemo_automodel.components.loggersnemo_automodel.components.lossnemo_automodel.components.modelsnemo_automodel.components.models.biencodernemo_automodel.components.models.commonnemo_automodel.components.models.deepseek_v3nemo_automodel.components.models.glm4_moenemo_automodel.components.models.gpt_ossnemo_automodel.components.models.llamanemo_automodel.components.models.qwen2nemo_automodel.components.models.qwen3_moenemo_automodel.components.models.qwen3_nextnemo_automodel.components.models.qwen3_omni_moenemo_automodel.components.models.qwen3_vl_moenemo_automodel.components.models.gpt2
nemo_automodel.components.moenemo_automodel.components.optimnemo_automodel.components.quantizationnemo_automodel.components.trainingnemo_automodel.components.utils
nemo_automodel.shared