NovaSky-AI/SkyRL
SkyRL: A Modular Full-stack RL Library for LLMs
What it solves
SkyRL is a modular full-stack library designed to make reinforcement learning (RL) for Large Language Models (LLMs) more accessible and efficient. It specifically addresses the challenges of training agents that can handle long-horizon, real-world tasks involving multi-turn tool use, such as coding, SQL generation, and terminal interaction.
How it works
SkyRL is organized into several specialized components:
- skyrl: A unified library for RL training on local hardware, combining a modular training framework (
skyrl-train) and a cross-platform backend for the Tinker API (skyrl-tx). - skyrl-agent: An agent layer focused on optimizing and scaling pipelines for multi-turn tool use LLMs on long-horizon tasks.
- skyrl-gym: A collection of RL environments implemented using the Gymnasium API, providing tasks for math, coding, search, and SQL.
Who it’s for
This library is designed for AI researchers and developers building RL-tuned LLMs, specifically those focusing on agentic workflows, tool-use capabilities, and long-horizon problem solving.
Highlights
- Tinker API Support: Implements a backend for the Tinker API, allowing users to run training scripts written for that API on their own GPUs.
- Modular Architecture: Provides a separate training framework, agent layer, and environment library (Gymnasium API).
- Longe-Horizon Focus: Optimized for real-world tasks like SWE-Bench and terminal-use agents.
- On-Policy Distillation: Supports advanced RL techniques such as on-policy distillation and fully async RL with in-flight weight updates.
Related
- Project
- Project
- Project
- Project
- Project