Gemma-4-125B-A12B
Gemma-4-125B-A12B is an expanded sparse Mixture-of-Experts language model based
on google/gemma-4-26B-A4B-it. This release focuses on agentic coding,
repository understanding, multi-turn tool use, explicit reasoning, and
long-context software tasks.
The model is released in MXFP4 format for the expert weights, with shared and
non-expert weights kept in BF16.
Model Summary
- Base lineage:
google/gemma-4-26B-A4B-it
- Architecture: sparse Mixture-of-Experts language model
- Expert layout:
688 total experts
- Active experts per token:
50
- Total logical text parameters: approximately
125B
- Active parameter class: approximately
A12B
- Weight format: MXFP4 experts with BF16 shared weights
- Intended serving mode: Gemma 4 chat, thinking, and tool-use template enabled
- Created on a two-GPU workstation
Expert Capacity
This checkpoint expands the Gemma 4 expert pool while preserving sparse
inference. Each token activates a selected subset of experts rather than the
full parameter set.
- Expert pool size:
688
- Active expert budget:
50 experts per token
- Active expert fraction per layer: approximately
7.27%
- Approximate logical active text size:
11.4B
- Approximate padded serving active size:
12.2B
Recommended Runtime
This model was created on a two-GPU workstation. The following command is the
tested two-GPU serving configuration:
1CUDA_VISIBLE_DEVICES=0,1 vllm serve /path/to/gemma-4-125b-a12b \
2 --served-model-name vllm/doobee \
3 --host 0.0.0.0 \
4 --port 23333 \
5 --dtype bfloat16 \
6 --tensor-parallel-size 2 \
7 --enable-expert-parallel \
8 --max-model-len 200000 \
9 --gpu-memory-utilization 0.96 \
10 --trust-remote-code \
11 --reasoning-parser gemma4 \
12 --tool-call-parser gemma4 \
13 --enable-auto-tool-choice \
14 --default-chat-template-kwargs '{"enable_thinking": true}' \
15 --language-model-only \
16 --skip-mm-profiling \
17 --max-num-seqs 1 \
18 --max-num-batched-tokens 8192 \
19 --enable-log-requests
Use a vLLM build with Gemma 4 MXFP4 MoE support. Long-context serving is
memory-intensive; the command above is configured for a 200k token context on
two high-memory GPUs.
Chat And Tool Use
The included chat template is intended to be used with thinking enabled.
Tool calling should be exercised through native OpenAI-compatible tool-call
paths rather than raw text parsing.
Recommended evaluation settings:
- Use
temperature=0.0 for deterministic smoke tests.
- Use
temperature=0.2 to 0.7 for normal agentic evaluation.
- Keep thinking enabled for the intended behavior profile.
- Use the included chat template and tokenizer files as shipped.
Intended Uses
- Agentic coding and software engineering tasks
- Repository exploration and codebase analysis
- Multi-turn tool-use workflows
- Long-context reasoning over technical material
- Patch planning, debugging, and implementation assistance
Limitations
- This is a large sparse MoE model and requires an inference stack that supports
Gemma 4 MXFP4 MoE serving.
- The model is optimized for tool-oriented assistant workflows and may not be
appropriate for all general-purpose chat settings.
- Long-context behavior depends heavily on serving configuration, GPU memory,
and request batching.
Credits
Thanks to NVIDIA for providing a broad range of pretraining and post-training
resources that helped make this work possible.