models

Modules

modelopt.models.arctic	Snowflake Arctic specs (trust-remote-code model type arctic).
modelopt.models.dbrx	DBRX specs (HF model type dbrx).
modelopt.models.deepseek	DeepSeek-MoE specs (trust-remote-code model type deepseek).
modelopt.models.gemma	Gemma 1/2/3 specs (HF model types gemma/gemma2/gemma3).
modelopt.models.gemma4	Gemma4 specs (HF model type gemma4).
modelopt.models.gpt_oss	GPT-OSS specs (HF model type gpt_oss).
modelopt.models.llama	Llama specs (HF model type llama).
modelopt.models.mixtral	Mixtral specs (HF model type mixtral).
modelopt.models.nemotron	Nemotron specs (HF model type nemotron); Nemotron-H lives in nemotron_h.py.
modelopt.models.nemotron_h	Nemotron-H specs (HF model type nemotron_h).
modelopt.models.qwen2_moe	Qwen2-MoE specs (HF model type qwen2_moe).
modelopt.models.qwen3	Qwen3 (dense) specs (HF model type qwen3).
modelopt.models.qwen3_5_moe	Qwen3.5-MoE specs (HF model type qwen3_5_moe).
modelopt.models.qwen3_moe	Qwen3-MoE specs (HF model type qwen3_moe).
modelopt.models.qwen3_next	Qwen3-Next specs (HF model type qwen3_next).
modelopt.models.registry	Registry indexing the per-model ModelSpec by HF model type.
modelopt.models.specs	Per-model descriptor classes.

Per-model descriptors, one module per HF model type.

Module names mirror the transformers.models layout (https://github.com/huggingface/transformers/tree/main/src/transformers/models); trust-remote-code models (arctic, deepseek) use their config model_type. Each model module registers one global ModelSpec (see specs.py); importing this package registers them all. Consumers resolve a spec via the registry lookups (get_spec / match_moe_block) and read its fields; an unmatched lookup returns None so callers fail loudly or fall back per their own policy.