APEX Coordinator — Qwen2.5-1.5B

A small learned coordinator that plans a team of LLM agents on long-horizon professional-services tasks (the APEX-Agents benchmark). Given a task it emits one plan: a schedule of which agents run in which round (parallel vs serial) and the routing between them:

{"batches": [["corpus_librarian", "metric_extractor"], ["ratio_calculator"]],
 "routes": {"ratio_calculator": ["corpus_librarian"]}}

Training: Stage-1 SFT warm-start (behaviour-cloning a commitment-parallel plan) → Stage-2 GRPO with a floor-protected reward (each sampled plan is executed by the real agent team + graded by a rubric judge).

Results (64-task financial test set, same grader)

system (workers) score serial-depth
single-agent 0.29–0.32 1.0
sequential 0.71–0.73 4.4
trained_coordinator (this model) 0.760 (Opus) / 0.795 (GLM) 3.2

Highest accuracy of the compared systems, and beats the sequential baseline on both accuracy and critical-path depth. A 5-seed reproducibility run: 0.760 ± 0.035 (Opus) / 0.795 ± 0.016 (GLM).

Load

from transformers import AutoModelForCausalLM, AutoTokenizer
tok = AutoTokenizer.from_pretrained("arcAman07/apex-coordinator-qwen1.5b")
model = AutoModelForCausalLM.from_pretrained("arcAman07/apex-coordinator-qwen1.5b")

It is used inside a multi-agent harness that executes the emitted plan with worker agents (Opus / GLM) and grades the result. The coordinator does greedy decoding of a single plan per task.

Downloads last month
328
Safetensors
Model size
2B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for arcAman07/apex-coordinator-qwen1.5b

Finetuned
(1769)
this model