This document is relevant for: Trn2, Trn3
GPT-OSS Model Recipe#
Introduction#
gpt-oss is an open-weight Mixture-of-Experts (MoE) language model family released by OpenAI, designed for strong reasoning, tool use, and long-context generation. The published checkpoints ship with MXFP4-quantized expert weights.
gpt-oss is supported for inference serving with
vLLM using the Neuron SDK on AWS
Trainium2 (trn2) and Trainium3 (trn3) hardware.
Compatible model checkpoints:
Model |
HuggingFace |
Hardware |
Quantization |
|---|---|---|---|
gpt-oss-20b |
Trn2, Trn3 |
BF16 (Trn2), MXFP4 (Trn3) |
|
gpt-oss-120b |
Trn2, Trn3 |
BF16 (Trn2), MXFP4 (Trn3) |
On Trn3, MXFP4-quantized weights are auto-selected from the published checkpoints. On Trn2, both models run in BF16. MXFP4 is Trn3 only.
Features#
Per-model feature availability for gpt-oss. See the features guide for configuration details and the cross-model feature compatibility matrix.
Category |
Feature |
Status |
|---|---|---|
Inputs |
Text |
✅ |
Quantization |
MXFP4 weights (Trn3) |
✅ |
BF16 weights (Trn2) |
✅ |
|
Parallelism |
Tensor parallelism (TP) |
✅ |
Data parallelism (DP) |
✅ |
|
Expert parallelism (EP) |
✅ |
|
Pipeline parallelism (PP) |
❌ |
|
Performance |
Continuous batching |
✅ |
Segmented prefill |
✅ |
|
Prefix caching (APC) |
✅ |
|
Speculative decoding (EAGLE3) |
✅ |
|
Disaggregated inference (1P1D / xPyD) |
✅ |
|
On-device sampling (greedy, top-k, top-p) |
✅ |
|
Serving |
Structured outputs / tool calling |
✅ |
Compilation |
torch.compile (XLA backend) |
✅ |
CPU mode (testing) |
✅ |
Status legend:
✅ Supported: integrated and tested for gpt-oss
❌ Not supported: may be considered for future releases
The deployment tutorial walks through deploying gpt-oss 20B and 120B on Trn3 (MXFP4) or Trn2 (BF16) via two paths: a single-instance server and a disaggregated-inference (separate prefill and decode) deployment.
Accuracy Evaluation#
Accuracy measured on real hardware with MXFP4 weights (Trn3), medium reasoning effort.
Benchmark |
gpt-oss-120b (MXFP4) |
|---|---|
GSM8K-CoT |
88.8% |
AIME25 (avg@8, medium) |
78.75% |
GPQA-diamond (medium) |
72.22% |
Tutorials#
Tutorial: Deploy gpt-oss with vLLM Neuron — End-to-end single-instance and disaggregated-inference deployment recipe for gpt-oss 20B and 120B.
Prefix caching benchmark tutorial — Benchmark prefix caching with gpt-oss across open-source datasets.
This document is relevant for: Trn2, Trn3