This repository presents a recursive reinforcement learning pipeline where a large AI agent is trained to design and execute training jobs for smaller models. The system utilizes a dual-loop RL architecture where the performance of the trained small model serves as the reward signal for the larger trainer agent.
Highlights
Implements a dual-loop RL architecture (Outer loop for the trainer, Inner loop for the target model)
Fully open-sourced, including LoRA weights, reward code, and GPU orchestration scripts
Demonstrates successful skill transfer to held-out task families the agent had not previously encountered
Leverages Tinker and GRPO for the reinforcement learning processes