Trace is the Next AutoDiff: Generative Optimization with Rich Feedback, Execution Traces, and LLMs Allen Nie

Neural Information Processing Systems 

We study a class of optimization problems motivated by automating the design and update of AI systems like coding assistants, robots, and copilots. AutoDiff frameworks, like PyTorch, enable efficient end-to-end optimization of differentiable systems. However, general computational workflows can be non-differentiable and involve rich feedback (e.g.