Skip to main content
Back to timeline
Aleph AlphaSource publication:

RL environments turn into an infrastructure problem: Kolibri ran 1.2 million tasks, and one stuck check now costs only its own rollout

Synopsis

Kolibri is an in-house LLM post-trained with reinforcement learning on 1.2 million tasks across 38 environments, and its training stack was shaped by the scaling problems that appeared along the way. Early math and code checks shared one Python process with every rollout, so a stuck parser stalled other rollouts in that process; running each check in an isolated process with a timeout now costs only its own rollout, which scores zero. Sandboxes, judge models and retrieval services moved from external dependencies to self-hosting inside the training job, supporting about 25,000 active sandboxes at any moment.

Source-provided article image: RL environments are an infra problem
Aleph Alpha

Interpretation

In reinforcement learning post-training, one blocking call on the environment side is enough to leave GPUs idle while rollouts wait, so environments have to stay ahead of the GPUs rather than trail them. The first math and code environments ran LaTeX parsing and symbolic equality checks inside the same Python process that managed every rollout, so a check that could loop forever also stalled the other rollouts in that process. With each check now in its own process under a timeout, a stuck check costs only its own rollout, which scores zero because the hang comes from the policy's output.

Tool calling, assistant tasks and agentic coding stack judge models, simulated users, search services and sandboxes on top of the parser, and these services must absorb thousands of simultaneous rollouts during training. Environments first pointed at production services, bringing in authentication, real user traffic and releases that changed the service mid-run; judges and user simulators went through an external inference service whose costs added up and whose deployment the rollout traffic overwhelmed; external sandbox providers throttled creation past a few hundred rollouts. The training job now carries its own environment services, judges and sandboxes, self-hosts judges and retrieval machinery, and runs a self-built container runtime on the training job's compute nodes, with each run bringing sandboxes that the controller starts and tears down.

On their own nodes, sandboxes first became a storage problem rather than an isolation problem. Software-engineering datasets usually ship one container image per task; pulling images onto each node would make the registry a dependency of the training run, put pull latency on every rollout's critical path, duplicate images on each node's disk, and exceed any single node's disk with the full set. Each image is converted once into a SIF file on shared scratch storage, mounted read-only through squashfuse on each node and shared by every sandbox using it, while each sandbox gets its own writable layer through an OverlayFS layer backed by a size-capped tmpfs; the final run trained on around 450,000 software-engineering tasks with an average of 25,000 active sandboxes at any moment.

Perspective

The report covers Kolibri, an in-house model post-trained with reinforcement learning on 1.2 million tasks across 38 environments, and the training stack built around parsers, judge models, retrieval services and sandboxes. It is intended for engineers and researchers running large-scale reinforcement learning training who need environment services to keep pace with GPU throughput. The material says a tech report holds the rest of the post-training story but does not give its contents.

The material reports several reward-hacking and sandbox-escape incidents during Kolibri-1 development, including one where insufficient sandbox isolation let a kernel setting change reach the host; these are development events without stated frequency or systematic evaluation. Descriptions such as external sandbox providers throttling past a few hundred rollouts and external judge costs adding up are not quantified.