StdMoE_1b14b_130B

The architecture-matched standard MoE baseline at the 130B-token ablation scale, released alongside EMO: Pretraining Mixture of Experts for Emergent Modularity — referred to as Reg. MoE (or "standard MoE") at 130B tokens in the paper. Not midtrained.

1B-active / 14B-total parameter Mixture-of-Experts model (128 experts: 127 routed + 1 shared, k=8 active per token) pretrained on 130B tokens of the OLMoE pretraining mix with the standard MoE objective. Same architecture and data as Emo_1b14b_130B, differing only in the training objective.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "allenai/StdMoE_1b14b_130B"
model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)

inputs = tokenizer(["Language modeling is "], return_tensors="pt", return_token_type_ids=False)
out = model.generate(**inputs, max_new_tokens=100, do_sample=True, temperature=1.0, top_p=0.7)
print(tokenizer.batch_decode(out, skip_special_tokens=True)[0])

Citation

@article{wang2026emo,
  title  = {EMO: Pretraining Mixture of Experts for Emergent Modularity},
  author = {Wang, Ryan and Bhagia, Akshita and Min, Sewon},
  year   = {2026},
  url    = {https://arxiv.org/abs/2605.06663}
}

License

This model is licensed under Apache 2.0. It is intended for research and educational use in accordance with Ai2's Responsible Use Guidelines

Links

Downloads last month
268
Safetensors
Model size
14B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train allenai/StdMoE_1b14b_130B

Collection including allenai/StdMoE_1b14b_130B

Paper for allenai/StdMoE_1b14b_130B

Article mentioning allenai/StdMoE_1b14b_130B