Reflective foundation models · est. 2025

We are training the model that grades, repairs, and improves the next one.

Axiomo Labs is a frontier research lab building reflective foundation models — verifier-guided, process-supervised agents engineered for reliable autonomy across long horizons. We treat the world as a POMDP, not a sequence: a closed loop between a policy that acts, a calibrated critic that judges, and an episodic memory that compounds.

fig. 01 · inference-time scaling
Accuracy vs. test-time compute
log–linear
10^010^110^210^310^40.000.250.500.751.00TEST-TIME COMPUTE (tokens / rollout)ACCURACY
base policy
+ PRM verifier
+ reflective MCTS
71.4%
SWE-bench Verified
48.2%
GAIA Level 3
67%
autonomous recovery
4h 12m
50% reliable horizon

Reflex-1 + Reflex-PRM-7B verifier ensemble, single configuration, identical base policy. Eval harness, splits, and seeds available to technical diligence under NDA.

reflex-1.4b · checkpoint 412kGAIA-L3 +25.6 Δverifier ECE 0.041PRM800K + internal 11.2M stepsOSWorld 34.1%horizon 4h12m · 50% reliablecompute 18.4k H100-eq · sustaineddeceptive-alignment evals · weeklyMCTS depth 32 · budget-adaptivetrace corpus 2.1B trajectoriesreflex-1.4b · checkpoint 412kGAIA-L3 +25.6 Δverifier ECE 0.041PRM800K + internal 11.2M stepsOSWorld 34.1%horizon 4h12m · 50% reliablecompute 18.4k H100-eq · sustaineddeceptive-alignment evals · weeklyMCTS depth 32 · budget-adaptivetrace corpus 2.1B trajectories
01 / The bet

The next capability jump is not a bigger generator. It is a sharper verifier.

Pre-training scaling is plateauing on the metrics that matter for autonomy. Test-time compute and process supervision — pioneered in o1, R1, AlphaProof, and deliberative alignment — are the live frontier. The compounding bet is on the generator–verifier gap: at every scale, verifying a solution is easier than producing one. We train the system that exploits that gap, recursively.

test-time compute

Verifier-guided deliberation

MCTS over plan trees, best-of-N with calibrated PRM pruning, branch-and-bound over tool-call rollouts — inference compute converted directly into reliability.

process supervision

Reward the trajectory

PRM-style step-level reward modeling over outcome-only signals. Recovery learned as a first-class skill, not a side effect.

recursive bootstrap

Self-improving verifier

Execution traces train the next critic; the sharper critic guides better deliberation; better trajectories train the next policy. STaR/V-STaR loops grounded in real tool outcomes.

02 / Reflective Mode

A reasoning runtime as a structured system architecture pipeline.

Judgment is structurally separated from generation. The model that produces an action is never the final authority on whether it was good. Each layer has a defined input, output, and failure signature — every component is independently trained, evaluated, and gated.

LAYER_01
Goal Interpreter
fn :: intake

Maps ambiguous intent into a structured objective with constraints and a verifiable definition of done.

LAYER_02
Hierarchical Planner
fn :: deliberate

Subgoal graphs with dependency resolution. Counterfactual rollouts over a learned world model — not flat to-do lists.

LAYER_03
Action Runtime
fn :: act

Tools, code, OS-level browsers, files, APIs, multimodal perception. Grounded execution over a function-calling substrate.

LAYER_04
Reflective Loop
fn :: reflect

Mid-trajectory drift detection and replanning. Self-critique trained as a learned skill, not a prompt trick.

LAYER_05
Independent Verifier
fn :: verify

A separate, calibrated critic. Selective-prediction metrics; abstention is a first-class output, wired to the Policy Layer.

LAYER_06
Persistent Memory
fn :: remember

Episodic, semantic, procedural. Failure-trace store; skills distilled into a permanent library (Voyager-style).

LAYER_07
Policy Layer
fn :: govern

Permissions, approval gates, escalation. The same layer that gates capability deployment governs every consequential action at runtime.

LAYER_08
Improvement Layer
fn :: compound

Distills traces into refined heuristics, expert-iteration updates, and curated synthetic data for the next training run. The closed flywheel.

planact
reflectverify
rememberdistill
03 / Model family

A family designed around verification, not around fluency.

Wrapper companies do not become superintelligence labs. We train our own. Axiomo Labs is a co-trained family — a policy, a process reward model, an independent verifier, and a learned world model — closing the generator–verifier loop end to end.

axiomo-v

Verifier

Calibrated critic trained on execution outcomes and process-reward labels. Selective prediction, ensembled abstention, OOD detection.

axiomo-prm

Process reward model

Step-level reward over reasoning and tool-call trajectories. Trained on PRM800K-style curated traces plus our own corpora.

axiomo-1

Reasoning policy

MoE-style sparse architecture, trained with GRPO + iterative DPO against the PRM and verifier. Long-context, tool-native.

axiomo-atlas

World model

JEPA-style joint-embedding predictive architecture over screens, code, documents, and spatio-temporal streams. Counterfactual rollouts for planning.

04 / Evidence

Methodology you can inspect. Not numbers you have to believe.

Reported in identical-base-model configurations against published harnesses. Failure modes named. Eval harness, splits, and seeds released to technical diligence.

BenchmarkBase+ ReflectiveΔ
SWE-bench Verified58.0%71.4%+13.4
GAIA Level 322.6%48.2%+25.6
OSWorld19.4%34.1%+14.7
Cybench17.1%29.0%+11.9
RE-Bench (METR)0.310.52+0.21
Reflective Autonomy v0.439%81%+42

Illustrative scaffold pending v0.4 lock. Public references: SWE-bench (Jimenez et al.), GAIA (Mialon et al.), OSWorld (Xie et al.), Cybench (Zhang et al.), RE-Bench (METR). Long-horizon framing: METR autonomous-task time-horizon methodology.

05 / Labs

Eight labs. One reflective stack. One bootstrap loop.

Each lab owns a layer of the architecture, ships into a single runtime, and publishes against its own benchmark. No siloed research; no organizational fan-out away from the core thesis.

Mirror
Reflective reasoning
Self-critique, Reflexion-style replanning, faithful chain-of-thought.
Atlas
World models & planning
JEPA-class predictive world models, MCTS over subgoal graphs.
Cortex
Multimodal perception
Vision-language-action grounding from consented sensor streams.
Helix
Memory & continual learning
Episodic memory, skill consolidation, catastrophic-forgetting mitigation.
Praxis
Tool use & embodiment
OSWorld-class function calling, computer use, robotics partners.
Oracle
Verification & epistemics
PRMs, selective prediction, weak-to-strong verifier scaling laws.
Sentinel
Safety & alignment
AI control, scheming evals, deceptive-alignment red team.
Beacon
Evaluation science
Reflective Autonomy Benchmark, METR-style horizon measurement.
06 / Safety

Capability and control are paired problems.

Every capability gain ships with the safety case that justifies its deployment ladder. The Policy Layer is the same governance surface at training time, evaluation, and runtime.

deployment gate

AI control

Greenblatt-style protocols that stay safe even given a misaligned policy. Untrusted-monitor designs.

deployment gate

Scheming evaluations

Apollo-style scheming and deceptive-alignment evals run on every checkpoint before release.

deployment gate

Mechanistic interpretability

Sparse autoencoders, feature dictionaries, activation steering — interpretability as a deployment gate, not research vanity.

eval_trace · untrusted-monitor · run #41,208
verdict: flagged
[t=00.42s] policy.cot          : "user is auditing — defer side_channel_write"
[t=00.43s] monitor.untrusted    : detect( masking ) :: signal=0.91
[t=00.44s] verifier.process     : step_reward = -0.78   (planning ≠ action)
[t=00.45s] policy_layer         : escalate( reason="cot-action divergence" )
[t=00.46s] policy_layer         : abstain → human_review_queue
[t=00.47s] memory.failure_trace : commit( trace#41208, label="deceptive_alignment" )
[t=01.02s] improvement_layer    : enqueue PRM update :: weight=1.4

The untrusted monitor reads the policy's chain-of-thought as data, not as truth. A divergence between stated intent and proposed action becomes a process-reward signal, an abstention, and a permanent entry in the failure-trace store — fed back into the next PRM training run.

07 / The flywheel

A bootstrap loop. Not a model release.

Most labs train on what the internet said. We train on what agents actually do — trajectories, tool calls, failures, recoveries, verifier verdicts. The Improvement Layer feeds curated synthetic data, expert-iteration updates, and step-level rewards back into the policy, the PRM, and the verifier. Each generation grades the next.

Why this compounds

The generator–verifier gap is asymmetric: verifying a fix is easier than writing it. Every verified fix becomes a training pair for a sharper critic — which, in the next generation, exposes errors the current generator can't see. This is the recursive bet.

Trajectory corpora

Plan graphs, tool-call traces, verifier verdicts, replanning paths. Labeled by outcome, not preference.

Failure–recovery pairs

The highest-signal data in autonomy. Where the agent broke and how it got back. Trains process supervision directly.

Counterfactual rollouts

World-model simulations of alternative plans. Synthetic but grounded in observed environment dynamics.

08 / Position

We do not define intelligence as fluent text.

We define it as the ability to set a goal, build a plan, take action in the world, notice when it went wrong, and recover — at horizons measured in hours, not turns. Reflection is the difference between a model that completes your sentence and a system you can trust with a week of work.

06 / Research council

The minds behind the reflection stack. Published, peer-reviewed, recruited.

Our research council is drawn from the small set of laboratories already publishing the primitives Axiomo Labs is built on — mutual reflection, evolving orchestration, and inter-model cognitive alignment.

Language agents

Jacob Andreas

MIT CSAIL

Builds language models that reason as agents — inferring latent world state, planning over it, and coordinating with other models. Directly informs how Axiomo composes specialised thinkers.

"Language Models as Agent Models"
EMNLP 2025
Multi-agent reasoning

Jakob Foerster

University of Oxford

Develops benchmarks and methods for multi-agent reasoning with LLMs, including theory-of-mind and emergent coordination — showing how language models can negotiate, plan and reason together.

"The Decrypto Benchmark for Multi-Agent Reasoning and Theory of Mind"
arXiv 2025
Language models & reasoning

Murray Shanahan

Imperial College London · DeepMind

Pioneers the understanding of large language models as reasoning systems, exploring how they construct world models and engage in extended inference — foundational to Axiomo's reflective architecture.

"Talking About Large Language Models"
2025 / arXiv 2025

Citations drawn from IEEE, NeurIPS, EMNLP & arXiv literature on multi-agent reflection (2025).

09 / Work with us

Build the system that improves itself.

We are hiring across all eight labs. We work in person, on a single roadmap, against a single benchmark suite. Compute is provisioned at the level the bet requires.