Lakshya A Agrawal

AI PhD Student @ UC BerkeleyLaude Open Research ResidentAmazon AI PhD FellowCreator of GEPA & multilspyPreviously AI4Code at Microsoft Research

Lakshya A Agrawal

I am a Computer Science PhD student at UC Berkeley, advised by Prof. Matei Zaharia and Prof. Dan Klein. I am affiliated with Sky Lab, BAIR, and Berkeley NLP. I build AI systems that improve through experience, by optimizing their prompts, agent harnesses, and model weights.

My background in programming-language tools led to Monitor-Guided Decoding (MGD) at Microsoft Research. MGD built a static-analysis harness around an LLM to guide code generation, letting a smaller model outperform a larger one on code generation. This motivated my work at Berkeley on evaluating LLM systems with harnesses, rather than LLMs alone. In LangProBe, we mapped their cost–quality frontier and showed that the principle holds across many domains: smaller models with well-tuned harnesses can outperform much larger LLMs at lower cost.

This motivated exploring automated harness optimization from rich textual feedback. GEPA optimizes prompts by reflecting on execution traces and textual feedback, outperforming GRPO with fewer rollouts in our experiments; optimize_anything extends this to the full agent harness; and gskill learns repository-specific skills for coding agents. This led to studying how learning in text space could also support model training. mmGRPO is an RL method for multi-component LLM systems, combining weight updates with prompt optimization. Fast-Slow Training interleaves fast prompt adaptation with slower RL updates, so prompts absorb task-specific lessons while weights need not change as much. This improves data efficiency and performance while preserving the capacity of the model to learn subsequent tasks. I am continuing to study how self-improving AI systems should learn in the harness and in the model.

UpdatesRecent talks, releases, and milestonesView archive
01 / Research

Publications

Selected publications · 7 of 14
GEPA, MIPROv2, and GRPO compared by rollout budget on HotpotQA
Harness optimization

GEPA: Reflective Prompt Evolution Can Outperform Reinforcement Learning

Lakshya A Agrawal, Shangyin Tan, Dilara Soylu, Noah Ziems, Rishi Khare, Krista Opsahl-Ong, Arnav Singhvi, Herumb Shandilya, Michael J Ryan, Meng Jiang, Christopher Potts, Koushik Sen, Alexandros G. Dimakis, Ion Stoica, Dan Klein, Matei Zaharia, Omar Khattab

ICLR 2026 Oral

Reflective search that uses execution traces and natural-language feedback to improve an agent’s prompts.

Fast-Slow Training: fast context updates and slow model-weight updates learn from the same rollouts
Model training & RL

Learning, Fast and Slow: Towards LLMs That Adapt Continually

Rishabh Tiwari*, Kusha Sareen*, Lakshya A Agrawal*, Joseph E. Gonzalez, Matei Zaharia, Kurt Keutzer, Inderjit S Dhillon, Rishabh Agarwal, Devvrit Khatri

Preprint · 2026

Fast-Slow Training couples fast adaptation of prompts with slower reinforcement learning of model weights, enabling continual learning with less forgetting.

Details

* Equal contribution.

The optimize_anything loop connects an evaluator, diagnostic feedback, and an LLM proposer
Harness optimization

optimize_anything: Unified Text Optimization Can Outperform Specialized Systems

Lakshya A Agrawal*, Donghyun Lee*, Shangyin Tan*, Wenjie Ma, Karim Elmaaroufi, Rohit Sandadi, Sanjit A. Seshia, Koushik Sen, Dan Klein, Ion Stoica, Joseph E. Gonzalez, Omar Khattab, Alexandros G. Dimakis, Matei Zaharia

ACM CAIS 2026 Industry Spotlight Award

A unified approach to optimizing text, from agent architectures and code to scheduling policies and mathematical constructions.

Details

* Equal contribution.

A workshop version, optimize_anything: A Unified Framework for LLM-Based Mathematical and Scientific Optimization, appeared at the 3rd AI for Math Workshop at ICML 2026.

Slide from Noah Ziems’s ACM CAIS presentation comparing GRPO and multi-module GRPO
Model training & RL

mmGRPO: Composing Policy Gradients and Prompt Optimization for Language Model Programs

Noah Ziems*, Dilara Soylu*, Lakshya A Agrawal*, Isaac Miller, Liheng Lai, Chen Qian, Kaiqiang Song, Meng Jiang, Dan Klein, Matei Zaharia, Karel D'Oosterlinck, Christopher Potts, Omar Khattab

ACM CAIS 2026 Industry Spotlight Award

Multi-module GRPO trains compound language-model programs and composes reinforcement learning with prompt optimization.

Details

* Equal contribution.

MAST taxonomy groups failures into specification issues, inter-agent misalignment, and task verification
Understanding & evaluating agents

Why Do Multi-Agent LLM Systems Fail?

Mert Cemri*, Melissa Z. Pan*, Shuyi Yang*, Lakshya A Agrawal, Bhavya Chopra, Rishabh Tiwari, Kurt Keutzer, Aditya Parameswaran, Dan Klein, Kannan Ramchandran, Matei Zaharia, Joseph E. Gonzalez, Ion Stoica

NeurIPS 2025 · D&B Spotlight

MAST: a taxonomy of multi-agent failures, grounded in annotated execution traces, for understanding where and why agent systems break down.

Details

* Equal contribution.

LangProBe plots the cost and quality of models, programs, and optimizers
Understanding & evaluating agents

LangProBe: A Language Programs Benchmark

Shangyin Tan, Lakshya A Agrawal, Arnav Singhvi, Liheng Lai, Michael J Ryan, Dan Klein, Omar Khattab, Koushik Sen, Matei Zaharia

EMNLP 2025 · Findings

A benchmark of language-model programs that studies how models, architectures, and optimizers trade off quality and cost.

Open Source Projects

GEPA
Reflective optimization

GEPA & optimize_anything

Open-source software · Creator

A Python library for learning from rich textual feedback to optimize prompts, agent harnesses, and other text artifacts. optimize_anything extends the same reflective loop to any text artifact with a score, from agent architectures to CUDA kernels, and omni composes optimizers into a meta-optimizer.

6,759 GitHub stars60.8M PyPI downloads7.28M in the past month

DSPy project logo
Open-source infrastructure

DSPy

Core maintainer

A framework for programming with language models. My contributions include the GEPA and GRPO integrations and parts of the language-model interface.

CAL actor model showing input streams, internal state, actions, and output streams
Programming languages & systems

StreamBlocks GraalVM

EPFL · 2020–2021

A runtime for the CAL dataflow language on Truffle/GraalVM, with IDE and debugger support, developed in Prof. James Larus’s group.

Maxima symbolic computations and plotting, the host system for Pytranslate
Programming languages & systems

Pytranslate

Google Summer of Code · 2019

A Maxima-to-Python transpiler written in Common Lisp, including support for plotting. Distributed with Maxima since 2019.

Show moreShow fewer open-source projects
CHIP-8 emulator running a maze program

chip8emu

A terminal-based CHIP-8 emulator written in C++, for running classic games and other CHIP-8 programs.

CovidRelief Bot

A Telegram chatbot built to help find COVID-19 resource leads in India, including oxygen and hospital beds, and extract resource details from images.

Space_B__ars keyboard-catching gameplay

Space_B__ars

A Pygame game in which you capture escaped keyboard keys and combine them into commands to change the game.

CoVid lecture recording interface

CoVid

Browser-based lecture recording software designed to keep recordings small for students with limited internet connectivity.

Other open-source contributions include INRIA/spoon, mozilla/bugbug, and memorizing-transformers-pytorch.

02 / Sharing research

Talks & podcasts

EtsyGEPAGEPA
Etsy · AI Stitch · Aug 2026

From Prompts to Programs: Text Optimization with GEPA and DSPy

Netflix
Netflix Research · Jun 2026

GEPA, optimize_anything, and Fast-Slow Training

Blog posts

Honors & awards

Background

Education & experience

I interned at Ai2 (Allen Institute for AI) in Summer 2025. Before Berkeley, I was an AI4Code Research Fellow at Microsoft Research (2022–2024), working with Dr. Aditya Kanade, Dr. Navin Goyal, Dr. Shuvendu Lahiri, and Dr. Sriram Rajamani. I studied Computer Science and Applied Mathematics at IIIT-Delhi, graduating first in my department. I did my undergraduate thesis with Dr. Raghava Mutharaju, and spent two summers at EPFL with Prof. James Larus. I also interned as a software development engineer at Microsoft in 2021 and participated in Google Summer of Code in 2019, working on the Maxima-to-Python transpiler.

Teaching & service

I was a teaching assistant for Machine Learning (CSE343/ECE563) and Theory of Computation (CSE322) at IIIT-Delhi in 2021. I have reviewed for NeurIPS 2025, ICLR 2025, and ISEC 2024, and mentor student collaborators on the GEPA project.

Outside research

I grew up in Chhattisgarh and attended The Lawrence School, Lovedale. As a hobbyist, I play the saxophone, stylophone, Indian flute, and snare drums. I love classical music and jazz, films, and dancing.

Let’s talk.

I look forward to learning about the use cases and failure modes of AI systems, and the research questions they raise.

lakshyaaagrawal@berkeley.edu

Present & past affiliations

Present

Past