nemo_automodel.components.models.common

Common/shared components for model implementations.

Submodules

• nemo_automodel.components.models.common.bidirectional
• nemo_automodel.components.models.common.packing
• nemo_automodel.components.models.common.utils
• nemo_automodel.components.models.common.hf_checkpointing_mixin

Package Contents

Data

__all__

API

nemo_automodel.components.models.common.__all__

[‘HFCheckpointingMixin’, ‘HAVE_TE’, ‘HAVE_DEEP_EP’, ‘BackendConfig’, ‘get_rope_config’, ‘initialize_…