Agent JIT Compilation for Latency-Optimizing Web Agent Planning and Scheduling
Caleb Winston, Ron Yifeng Wang, Azalia Mirhoseini, Christoforos Kozyrakis
Abstract
Computer-use agents (CUAs) automate tasks specified with natural language such as "order the cheapest item from Taco Bell" by generating sequences of calls to tools such as click, type, and scroll on a browser. Current implementations follow a sequential fetch-screenshot-execute loop where each iteration requires an LLM call, resulting in high latency and frequent errors from incorrect tool use. We present agent just-in-time (JIT) compilation, a system that compiles task descriptions directly into executable code that may include LLM calls, tool calls, and parallelization. Our approach comprises three components: (1) JIT-Planner, which generates multiple code plans, validates each against tool specifications, and selects the minimum-cost candidate; (2) JIT-Scheduler, which explores parallelization strategies via Monte Carlo cost estimation from learned latency distributions; and (3) an invariant-enforcing tool protocol specifying precondition and postcondition requirements to reduce the rate of incorrect tool use. Across five applications, JIT-Planner achieves speedup and 28% higher accuracy over Browser-Use, while JIT-Scheduler achieves speedup and 9% higher accuracy over OpenAI CUA.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext af86aff9-45af-43bd-8a42-9af0cd242cb2Builds on4
- WebArena: A Realistic Web Environment for Building Autonomous AgentsShuyan Zhou, Frank F. Xu, Hao Zhu, Xuhui Zhou et al.ICLR 2024 · 1,197 citations
- Using an LLM to Help With Code UnderstandingDaye Nam, Andrew Macvean, Vincent J. Hellendoorn, Bogdan Vasilescu et al.ICSE 2024 · 264 citations
- OpenCUA: Open Foundations for Computer-Use AgentsXinyuan Wang, Bowen Wang, Dunjie Lu, Junlin Yang et al.NeurIPS 2025 · 151 citations
- WebAgent-R1: Training Web Agents via End-to-End Multi-Turn Reinforcement LearningZhepei Wei, Wenlin Yao, Yao Liu, Weizhi Zhang et al.EMNLP 2025 · 1 citation
Related papers
- Beyond Static Endpoints: Tool Programs as an Interface for Flexible Agentic Web ServicesMugeng Liu, Shuoqi Li, Yixuan Zhang, Yun MaICML 2026
- An LLM Compiler for Parallel Function CallingSehoon Kim, Suhong Moon, Ryan Tabrizi, Nicholas Lee et al.ICML 2024 · 142 citations
- From Imperative to Declarative: Towards LLM-friendly OS Interfaces for Boosted Computer-Use AgentsYuan Wang, Mingyu Li, Haibo ChenEuroSys 2026
- LLM-as-Scheduler: Agentic Workflow Dynamic SchedulingDawei Xiang, Kexin Chu, Wenyan Xu, Wenhui Zhang et al.ACL 2026
- AXIS: Efficient Human-Agent-Computer Interaction with API-First LLM-Based AgentsJunting Lu, Zhiyang Zhang, Fangkai Yang, Jue Zhang et al.ACL 2025 · 9 citations
