AIAny
Icon for item

TAO: Tool-Aware Joint Scheduling for LLM-Based Agentic Serving

Jointly decides when waiting agent sessions should run on GPUs and which node should serve them. Scheduling accounts for tool-return readiness, time-decaying node cache value, and token demand.

Introduction

Agent-serving schedulers face a timing problem that ordinary inference queues miss: a session waiting on an external tool may not yet be ready for its next GPU step. Treating tool readiness as part of scheduling can change which session should run now—and where—rather than simply prioritizing whichever request is waiting longest.

Key Findings
  • TAO combines expected near-term tool returns, node-specific cache credit that decays over time, and token footprint into a session value. This connects workflow state with GPU allocation rather than optimizing model calls in isolation.
  • A dynamic program jointly selects and places waiting sessions under capacity constraints. Joint decisions can account for competition among sessions and nodes, unlike independent placement heuristics.
  • Across 12 reported model, workload, and concurrency settings, the paper reports up to 26.8% lower latency, 36.3% lower mean time to first token, and 29.5% higher decode throughput versus the best evaluated baseline for each metric. These are maximum reported gains, not averages or guarantees across every setting.
How to Read the Results

The central idea is that an agent session’s usefulness to schedule depends on more than its queue position: whether its tool result is ready and whether a node can reuse relevant cached state both matter. The reported improvements are promising, but the abstract does not provide enough detail to assess the workload mix, hardware, baseline configurations, or variability. The figures should therefore be treated as author-reported results pending examination of the full evaluation.

Who Should Care

A good fit for researchers and serving-system engineers working on multi-step LLM agents, especially where tool execution and GPU inference share a service pipeline. Look elsewhere if the workload is ordinary single-turn inference with no meaningful tool waits or reusable node-local cache; the scheduling signals that motivate TAO may offer little benefit there.

Information

  • Websitearxiv.org
  • OrganizationsShanghai Jiao Tong University, Stanford University
  • AuthorsShuhuai Li, Shengdong Ye, Dongdong Ge, Yinyu Ye, Jianghao Lin
  • Published date2026/10/27

More Items

Estimates how committed tokens improve predictions at remaining masked positions, then uses those contributions to refine reinforcement-learning signals for diffusion language models. On LLaDA-8B-Instruct, it raises MATH500 and GSM8K scores by up to 9.0 and 5.3 points.

Coordinates world creation, trajectory exploration, and Harness evolution to improve multimodal agents. It organizes tasks by shared capabilities rather than surface categories, reporting mean gains of 34 points on training benchmarks and 26 on unseen benchmarks.

Lets a main agent synthesize executable workflows to coordinate and verify sub-agents as a task unfolds. The paper reports 88.08% overall accuracy in Forced mode across six benchmarks.