nemo_automodel.components.models

Convenience model builders for NeMo Automodel.

Currently includes: • build_gpt2_model – returns a GPT-2 causal language model (Flash-Attention-2 by default).

Subpackages

• nemo_automodel.components.models.deepseek_v3
  • nemo_automodel.components.models.deepseek_v3.layers
  • nemo_automodel.components.models.deepseek_v3.model
  • nemo_automodel.components.models.deepseek_v3.state_dict_adapter
• nemo_automodel.components.models.gpt_oss
  • nemo_automodel.components.models.gpt_oss.layers
  • nemo_automodel.components.models.gpt_oss.model
  • nemo_automodel.components.models.gpt_oss.state_dict_adapter

Submodules

• nemo_automodel.components.models.gpt2

Package Contents

Data

__all__

API

nemo_automodel.components.models.__all__

[‘build_gpt2_model’]