Dataset

Axion 1.0 Training Data

Contains 18,363 complete multi-turn text conversations across 2 SFT stages, plus 484 RL task records for scientific reasoning and tool-use training.

Hugging Face ↗Nov 2, 2026Code ↗

  • llm
  • nlp
  • rl
  • tool-use
  • scientific-reasoning
  • supervised-fine-tuning
Two-stage data construction pipeline diagram with examples and dataset counts.
Two-stage data construction pipeline diagram with examples and dataset counts.

What it is

18,363 scientific reasoning and tool-use SFT trajectories plus 484 RL tasks for training open agent models. Covers self-contained reasoning, Python use, and verified external lookup. 18,363 full text trajectories in 2 SFT stages plus 484 RL tasks: stage1 self-contained scientific reasoning with Python, stage2 verified tool-use from public databases, and RL prompts with reference answers and metadata.

Key results

  • 10,819 stage1 reasoning trajectories
  • 7,544 stage2 verified tool-use trajectories
  • 484 RL tasks; Apache-2.0; Parquet

Useful for SFT and RL training of agentic LLMs that reason, call tools, and answer scientific questions.

One useful ML email a week.

Worth your time, not all of your time.

We track new ML work across papers, code, and data, then keep only the pieces worth a closer look. Selected, not collected.

Papers.Data.Code

The Weekly ML Signal

Selected, not collected.

One useful ML email a week: the papers, repos and datasets worth your attention.

Worth your time, not all of your time.