nemo_automodel#
Subpackages#
nemo_automodel._diffusersnemo_automodel._transformersnemo_automodel.autonvtxnemo_automodel.componentsnemo_automodel.components._peftnemo_automodel.components.attentionnemo_automodel.components.checkpointnemo_automodel.components.checkpoint._backportsnemo_automodel.components.checkpoint._torch_backportsnemo_automodel.components.checkpoint.addonsnemo_automodel.components.checkpoint.checkpointingnemo_automodel.components.checkpoint.state_dict_adapternemo_automodel.components.checkpoint.stateful_wrappersnemo_automodel.components.checkpoint.utils
nemo_automodel.components.confignemo_automodel.components.datasetsnemo_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.mistral3nemo_automodel.components.models.nemotron_parsenemo_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
Package Contents#
Functions#
Lazily import and cache submodules listed in all when accessed. |
|
Expose the names of all available submodules for auto-completion. |
Data#
API#
- nemo_automodel._original_pin_memory_loop#
None
- nemo_automodel._original_pin_memory#
None
- nemo_automodel._original_pin_memory_sig#
‘signature(…)’
- nemo_automodel.__all__#
[‘recipes’, ‘shared’, ‘components’, ‘version’, ‘package_name’]
- nemo_automodel.__getattr__(name: str)#
Lazily import and cache submodules listed in all when accessed.
- Raises:
AttributeError if the name isn’t in all. –
- nemo_automodel.__dir__()#
Expose the names of all available submodules for auto-completion.