Skip to content

Supported Models

XoRL discovers model implementations through the architecture names in the checkpoint configuration. The table below reflects the classes registered by the current source tree.

Model familyRegistered Hugging Face architecture name(s)Checked-in example
DeepSeek V3DeepseekV3ForCausalLM
DeepSeek V4DeepseekV4ForCausalLM
GLM-4 MoEGlm4MoeForCausalLMexamples/local/dummy/configs/full/glm4_moe_ep8.yaml
GLM-5 / GLM MoE DSAGlm5ForCausalLM, GlmMoeDsaForCausalLM
GPT-OSSGptOssForCausalLMexamples/local/dummy/configs/full/gpt_oss_20b_ep8.yaml
LlamaLlamaForCausalLMexamples/local/dummy/configs/full/llama3_8b.yaml
MiniMax M3 sparseMiniMaxM3SparseForCausalLM, MiniMaxM3SparseForConditionalGeneration
Nemotron-HNemotronHForCausalLM
OLMo 2Olmo2ForCausalLM
Qwen2Qwen2ForCausalLM
Qwen3Qwen3ForCausalLMexamples/local/dummy/configs/full/qwen3_8b.yaml
Qwen3 MoE / Coder MoEQwen3MoeForCausalLMexamples/local/dummy/configs/full/qwen3_30b_a3b_ep8.yaml
Qwen3.5Qwen3_5ForCausalLM, Qwen3_5ForConditionalGenerationexamples/local/dummy/configs/full/qwen3_5_4b.yaml
Qwen3.5 MoEQwen3_5MoeForCausalLM, Qwen3_5MoeForConditionalGenerationexamples/server/configs/full/qwen3_5_35b_a3b_full.yaml

The registry is defined in src/xorl/models/registry.py. When architectures is a list, the loader selects its first entry (architectures[0]). It does not scan later entries for a supported architecture; if the first entry is not registered, loading fails and reports the registered names.

XoRL accepts Hugging Face-style checkpoints:

  • config.json containing an admitted architecture and model configuration
  • one or more *.safetensors weight files
  • tokenizer files such as tokenizer.json and tokenizer_config.json

Specify the checkpoint with model_path, using either a local path or Hugging Face Hub ID. Use config_path and tokenizer_path when configuration or tokenizer artifacts live elsewhere.

FieldDescription
model_pathLocal checkpoint path or Hugging Face Hub ID.
config_pathConfiguration path. Defaults to model_path.
tokenizer_pathTokenizer path. Defaults to config_path.
attn_implementationRequested attention backend. Availability is architecture-specific.
moe_implementationRequested MoE kernel, such as triton, native, quack, or eager. Availability is architecture-specific.

Concrete server examples currently cover Qwen3, Qwen3-MoE/Coder, Qwen3.5-MoE, and GPT-OSS across selected full-weight, LoRA, and QLoRA modes. Browse examples/server/configs/ for the exact filenames; do not infer an unlisted filename by changing a model size in another example.

For supported MoE families, model loading converts the source checkpoint representation into the fused expert layout required by the selected XoRL kernels. The conversion and admitted layouts are architecture-specific; no separate user preprocessing step is required for the checked-in examples.

See Mixture of Experts for expert_parallel_size, ep_dispatch, and MoE implementation options.