What problem does it solve?
Agent Lightning removes the complexity of training AI agents by turning execution traces, rewards, and rollout data into a structured optimization workflow. It helps teams move from ad hoc prompting to measurable agent improvement with reinforcement learning and prompt optimization.
Core Features & Use Cases
- Agent Instrumentation: Add lightweight emission hooks to capture inputs, outputs, tool calls, rewards, and execution spans from existing agents.
- Training Orchestration: Manage rollouts, runners, algorithms, and resource updates through LightningStore and the Trainer loop.
- Optimization Workflows: Support reinforcement learning, automatic prompt optimization, and supervised fine-tuning for agents built with Python or JavaScript.
- Observability and Debugging: Integrate OpenTelemetry, query traces, inspect rollout status, and validate training performance with metrics.
- Use Case: A team building a customer-support agent can trace every interaction, score responses with a reward function, and continuously improve prompts and policies across repeated training runs.
Quick Start
Use Agent Lightning to instrument your agent, connect it to a LightningStore, and run a training loop that collects traces and optimizes responses.