Joint Optimization of Tool Creation and Use for Large Language Model Agents
Published in NeurIPS 2025 Main, 2026
We propose SMITH (Schema-grounded Multi-task Iterative Tool Honing), a reinforcement learning framework that jointly trains tool creation and tool use within a single policy. A 4B model trained on 13 procedural reasoning tasks reaches 79.9 macro-average accuracy on held-out tasks, outperforming an untrained 30B tool-writer.
Recommended citation: Tam, Z.R., Lin, C.Y., Chen, Y.N., Sun, S.H., & Lee, H.Y. (2026). "Joint Optimization of Tool Creation and Use for Large Language Model Agents." NeurIPS.
Download Paper
