OPEN AGENT FOUNDATION MODELS

Axion 1.0

Strong agents. A simpler recipe.

Paper ↗Code Coming soonDataset Coming soonModel Coming soon
18Ktraining examples
35B / 122Bmain MoE models
128KSFT context window
≈1K122B training GPU-hours
01 / THE IDEA

Put the complexity
in the learning problem.

Building a capable agent does not have to mean building an increasingly elaborate training pipeline.

Axion 1.0 follows a clear progression: learn to reason, learn to acquire missing information, then refine those capabilities through interaction. The recipe combines two lightweight supervised fine-tuning stages with reinforcement learning.

The central investment is in the data: challenging, model-aware scientific problems and tool-dependent tasks with verifiable answers. This gives the model substantive problems to learn from at each stage.

A simple recipe for capable agents.
02 / RESULTS

Broad capability.
Measurable progress.

SIX EVALUATION SETTINGS

Axion 1.0 is evaluated alongside leading open and proprietary models across scientific reasoning, information seeking, and biological data analysis.

Hover or focus on a model to highlight it across all six benchmarks. Click or tap to keep it selected.

Blue: Axion 1.0-122B · Teal: Axion 1.0-35B · Gray: comparison models

View original paper figure ↗

Data from the paper’s comparison figure. Scores (%), as reported in the manuscript. HLE here is the full benchmark; training-stage comparisons below use HLE-130. BMB uses the human-difficult subset. Axion scores are the best single-run scores from three runs, or five for BMB. Baseline evaluation settings may differ; see the comparison table for source notes.

Compare with selected baselines +
Selected results from the paper's overall performance table
ModelHLEBMBSEAL-0HiPhOFS OlympiadFS Research
Agents-A147.6—56.446.479.040.0
Apodex 1.1 (ReAct)53.223.5———55.0
Apodex 1.1 (Agent Team)56.135.3———63.3
Gemini 3.1 Pro51.435.3*53.2*38.5*85.0*26.7*
Axion 1.0-35B52.235.362.245.786.056.7
Axion 1.0-122B59.041.758.646.287.066.7

* Evaluated by the Axion team. Other baseline scores are from official reports. Cross-source comparisons may use different tools or execution settings. A dash denotes an unreported result.

03 / THE RECIPE

Start with
the right data.

Problem construction and trajectory collection are separate. Establish a question and a verifiable answer first; collect correct solution trajectories second.

Two-stage data construction: model-aware self-contained reasoning tasks, followed by verified tool-dependent tasks

Model-aware task strengthening

Composition, parameterization, range shifts, and case analysis make problems harder in response to strategies the base model already uses successfully.

Verifiable tool-dependent tasks

Structured data sources establish reference answers during construction. Solvers must recover the needed evidence through tools, without direct database access.

Three stages.
One continuous progression.

No dedicated mid-training, DPO, or on-policy distillation stages. Hover or select a stage to explore its role.

01SFT / STAGE I

Learn to reason.

Build self-contained scientific reasoning from public seed problems, then strengthen the questions using feedback from the base model.

Challenging problems → verified reasoning
02SFT / STAGE II

Learn to investigate.

Extend reasoning to tasks that require external information. Construct questions from real data and deterministic computation, and retain verified tool-use trajectories.

Missing information → grounded interaction
03REINFORCEMENT LEARNING

Learn through interaction.

Refine long-horizon reasoning and tool use with asynchronous GRPO and session-level training across agent trajectories.

Agent trajectories → refined behavior
TRAINING PROGRESSION

From base model to agent.

Choose a benchmark, then hover or select a training stage to compare both models.

Scores (%) from the paper’s training-stage table. HLE-130 is a 130-question text-only subset, distinct from full HLE above; BMB uses the human-difficult subset. Sequential checkpoints show training progress; they do not isolate the causal effect of individual components.

04 / BUILT TO TRAIN

Long trajectories.
Practical infrastructure.

SFT preserves complete reasoning and tool-use trajectories within a 128K context window. RL runs asynchronously: rollout workers interact with tools while a separate trainer updates the policy.

7h 30m122B · two SFT stages240 GPU-hours on 32 GPUs
2h 48m35B · two SFT stagesApproximately 90 GPU-hours on 32 GPUs
≈15h122B · RL40 rollout + 16 training GPUs · 40 steps
≈30h35B · RL8 rollout + 8 training GPUs · 20 steps
01

Full trajectories for SFT

FSDP, expert parallelism, and context parallelism support training on long interaction histories.

02

Asynchronous agent rollouts

Dedicated rollout and trainer pools let environment interaction and policy updates proceed concurrently.

03

Consistency across execution

Token-In-Token-Out preserves token IDs; rollout routing replay aligns MoE expert choices between rollout and training.

05 / OPEN RESEARCH

An open recipe.
A reproducible process.

Share the methods and code behind the training process.

Our open research materials cover data construction, prompts, training code, and the evaluation protocol.

Data constructionPromptsTraining codeEvaluation protocol
06 / THE TEAM

Axion 1.0

Open Agent Foundation Models, Simplified

Yuqi Jia1 · Yunjie Tian2,† · Jiayu Zheng2
Yijun Liang3 · Jingchen Sun4 · Mengyu Zhao2 · Di Fu2
1 Duke University2 ByteDance3 University of Maryland, College Park4 University at Buffalo

† Project Lead

Read the abstract +

We introduce Axion 1.0, an agent foundation model that combines strong and broad agentic capabilities with a remarkably simple training recipe. Axion 1.0 is trained in only three stages: two lightweight supervised fine-tuning stages followed by reinforcement learning, without dedicated mid-training, direct preference optimization (DPO), or on-policy distillation (OPD).

The first SFT stage develops strong self-contained reasoning through model-aware data construction. The second stage extends these capabilities to tool-dependent reasoning through diverse and automatically verifiable tasks grounded in real-world data sources and deterministic computation. Finally, reinforcement learning further strengthens long-horizon reasoning and interaction with external tools and environments.

Axion 1.0 demonstrates that strong agentic capabilities can emerge from a simple training recipe, offering a practical path for further research on agent foundation models.

Adapted from the paper’s abstract.
Visitor map