Reinforcement Learning
Reinforcement learning on language models can require large numbers of sampled completions, but the total varies substantially with the algorithm, the number of generations per prompt, dataset size, training steps, stopping criteria, and the task itself. TRL's PPO trainer exposes total episode count as a configurable parameter, and its GRPO trainer exposes the number of generated completions, generation batch size, and update count, so rollout volume is a design decision rather than a fixed requirement. What is consistent is that sampling is repeated across many completions, so generated tokens can become the dominant cost in a rollout budget.

Small open source models ranging from roughly 360M to 4B parameters are therefore practical candidates for controlling rollout memory and compute. Paired with the right AI infrastructure, teams can run large parallel rollouts without over-provisioning GPUs. This guide examines seven entries for cost-effective RL experimentation in 2026: one execution platform, Modal, followed by six open-weight models. Modal is listed first because the platform choice shapes rollout economics as much as the model choice does; it is not a seventh model, and the six models below are not ranked against each other by a measured rollout benchmark.
Modal provides serverless AI infrastructure for RL training, rollout inference, secure environment execution, batch processing, and interactive development, with compute available across CPUs and GPUs. Rather than managing clusters or reserving capacity, teams deploy jobs against Modal's multi-cloud GPU capacity pool, which is available without quotas or advance reservations.
Modal combines the execution layer for both training and inference in a single platform:
Modal's infrastructure has measurable results in production AI workloads:
Modal states that it has successfully completed a SOC 2 Type II audit and is SOC 2 Type II compliant. Modal supports HIPAA-compliant workloads on Enterprise plans via a BAA. Compute jobs run containerized and virtualized using gVisor, which adds a user-space kernel boundary around workload syscalls, with user data encrypted in transit and at rest and public APIs using TLS 1.3, per Modal's security documentation. Best For: Teams that want elastic GPU capacity for RL training without capacity management overhead, and that benefit from training, inference, and sandboxed execution living on one platform.
Qwen3-4B is a strong candidate at the top of the small-model range for single-GPU RL fine-tuning work. The official card reports 4.0B total parameters and 3.6B non-embedding parameters; calculating from the published architecture and tensor dimensions gives approximately 4.02 billion parameters in a dense architecture. A note on hardware: parameter-efficient fine-tuning fits comfortably on a single 80GB GPU. Full-parameter mixed-precision AdamW training does not fit comfortably. At roughly 18 bytes per parameter for weights, gradients, and optimizer state, about 72.4GB is consumed before activations, CUDA context, and temporary tensors, leaving little headroom. Fitting it requires aggressive optimization such as optimizer-state quantization, activation checkpointing or offload, very short sequences, a microbatch size of one, or LoRA/QLoRA instead of full-parameter training. An RL workflow may additionally need policy, reference, reward, value, or rollout-serving components depending on the algorithm.
Results are labeled by evaluation mode, because thinking and non-thinking scores are not comparable. The figures below come from the comparison tables published in the SmolLM3 model card, which evaluates Qwen3 checkpoints in both modes alongside its own:
| Mode | AIME 2025 | GSM-Plus | LiveCodeBench v4 |
|---|---|---|---|
| No thinking | 17.1 | 82.1 | 24.9 |
| Extended thinking | 58.8 | 88.2 | 52.9 |
Thinking mode more than triples the AIME 2025 result (approximately 244% relative) and more than doubles LiveCodeBench (approximately 112% relative), while GSM-Plus moves by roughly 7% relative. The gain is real but highly benchmark-dependent, and it is paid for in generated tokens: the Qwen3 card recommends output allowances as large as 38,912 tokens for difficult math and programming evaluations. Best For: RL training that needs stronger mathematical and coding reasoning and can accommodate parameter-efficient fine-tuning on a single GPU. Run Qwen3-4B inference on Modal's GPU instances, selecting the GPU type and count that fit your rollout profile.
SmolLM3-3B offers one of the more transparent training recipes among competitive 3B models. Hugging Face published the training data mixture, code, configs, and intermediate checkpoints, which lets researchers inspect much of how the model reaches its performance. That disclosure is extensive but not yet complete: the model card states that the datasets used for mid-training and post-training will be uploaded later.
Mode-labeled results from the official evaluation tables:
| Mode | AIME 2025 | GSM-Plus | LiveCodeBench v4 |
|---|---|---|---|
| No thinking | 9.3 | 72.8 | 15.2 |
| Extended thinking | 36.7 | 83.4 | 30.0 |
On instruction following and tool use, the card reports 76.7 on IFEval and 92.3 on BFCL tool calling, both no-thinking results. Best For: Research teams that want a largely reproducible baseline with published pretraining data and intermediate checkpoints, while accounting for the pending mid-training and post-training data release. Deploy SmolLM3-3B through Modal's batch processing for large-scale rollouts.
Qwen3-1.7B is a practical entry point for rapid RL prototyping. The official card uses the 1.7B label; calculating from the published configuration gives approximately 1.72 billion parameters.
Mode-labeled results from the comparison tables in the SmolLM3 model card, which evaluates Qwen3 checkpoints in both modes:
| Mode | AIME 2025 | GSM-Plus | LiveCodeBench v4 |
|---|---|---|---|
| No thinking | 8.0 | 68.3 | 15.0 |
| Extended thinking | 30.7 | 79.4 | 34.4 |
Best For: Iterating on RL algorithms where training velocity matters more than maximum capability. Train Qwen3-1.7B on Modal with autoscaling tuned to your workload, using warm pools, buffers, and scaledown windows to match standing capacity to rollout demand.
OLMo-1B from Ai2 (formerly styled the Allen Institute for AI) prioritizes scientific reproducibility. The training data, training and evaluation code, intermediate checkpoints, and logs are publicly released, which lets researchers study training dynamics directly rather than inferring them. Two qualifications matter for this list. First, OLMo-1B-0724-hf is a base pretrained model, not an instruction-tuned checkpoint, so its out-of-the-box rollout behavior is not directly comparable to the instruct models elsewhere in this guide. Second, it is an older option: Ai2 now offers OLMo 2 1B as a newer small model and describes OLMo 3 as its latest family. The 0724 checkpoint remains appropriate where checkpoint density and exact historical reproducibility matter, but a current 1B shortlist should also evaluate OLMo 2 1B.
The published configuration specifies 16 layers, hidden size 2,048, intermediate size 8,192, a 50,304-token vocabulary, and untied input/output embeddings, implying approximately 1.28 billion actual parameters despite the 1B label. With Apache 2.0 licensing, it serves academic teams that need:
Best For: Academic and research RL work requiring complete reproducibility and training transparency, where a base model is acceptable. Run OLMo experiments on Modal notebooks for interactive GPU-backed development.
Qwen3-0.6B is the smallest entry in the Qwen3 lineup covered here and still retains thinking mode. The official rounded size is 0.6B; the published configuration implies approximately 596 million parameters.
Qwen3-0.6B is best treated as a fast iteration target rather than a capability play. Published comparable figures at this size are sparse: the Qwen3 technical report reports 24.74 on MMLU-Pro for the Qwen3-0.6B base model, which is a base-model result and not a thinking versus non-thinking comparison. If you need thinking-mode numbers at this size, generate them yourself under a stated harness, prompt, shot count, answer-extraction method, and checkpoint revision, since the official card does not publish a traceable thinking-mode IFEval or MMLU-Pro pair. Best For: Maximum iteration speed during RL algorithm development and educational use. Deploy through Modal's inference platform for production serving, where fast cold starts keep the iteration loop tight.
SmolLM2-360M-Instruct sits at the bottom of the size range covered in this guide. At 360 million parameters, it supports CPU inference and beats a larger comparator on several benchmarks, though not on all of them. Whether it is sufficient for a given RL task is an empirical question for your own reward signal, not a property of the size tier.
Under the model card's evaluation setup, the instruct checkpoint scores:
Note that the frequently quoted 54.5 HellaSwag and 53.0 ARC figures belong to the base SmolLM2-360M model, not the instruct checkpoint. Against Qwen2.5-0.5B-Instruct, SmolLM2-360M-Instruct is ahead on IFEval, HellaSwag, ARC, PIQA, and MMLU, and behind on MT-Bench, BBH, and GSM8K. It outperforms the larger comparator on several evaluations, not generally. Best For: Algorithm prototyping and educational RL work where compute cost must be minimized. Scale SmolLM2-360M rollouts through Modal's job queue, which supports autoscaling, retries, and up to one million pending asynchronous inputs.
Modal's serverless platform removes much of the infrastructure work that slows RL experimentation. The platform provides GPU access without quotas or reservations across a multi-cloud capacity pool, so teams can size rollouts to experiment demand rather than to pre-provisioned capacity. Modal describes bursting to more than 1,000 GPUs with headline timing measured in minutes, and seconds for certain clusters, with allocation shaped by hardware, topology, region, and supply.
Modal combines model training, inference serving, sandboxed execution, batch processing, and notebooks on a single platform, defined in code through SDKs in Python, TypeScript, and Go. RL workflows that run model inference, execute environment code, persist state, and train policy updates can use these integrated services and primitives, which reduces the amount of external infrastructure a team must operate.
Modal is engineered for fast cold starts and faster feedback loops, with an optimized filesystem that helps containers come online quickly without letting large images slow startup down, and Functions scale to zero by default when no inputs remain. For bursty rollout patterns this can meaningfully reduce idle spend, while min_containers, buffer_containers, and scaledown_window give teams direct control over how much warm capacity stays available between bursts. Modal documents the full cold start path, including queueing and first-invocation initialization, and Memory Snapshots can accelerate initialization-heavy Functions, often by 3 to 10 times according to Modal, with GPU Memory Snapshots extending the same approach to GPU state as an Alpha feature.
The platform includes primitives that map onto common RL patterns:
Modal states that it has completed a SOC 2 Type II audit, with compute isolation through gVisor containerization and virtualization and encryption in transit and at rest, per its security documentation. gVisor places a user-space kernel between workload syscalls and the host, which is what makes running LLM-generated code in a sandbox practical at scale. Modal supports HIPAA-compliant workloads on Enterprise plans via a BAA.
Modal reports powering infrastructure for more than 10,000 teams. Physical Intelligence reports 10 to 15 ms of added network overhead for remote robotic control inference, Runway uses Modal for real-time multi-node inference for Runway Characters, its real-time video-agent API, and Quora uses Modal Sandboxes at scale to run LLM-generated code for Poe. On the agent side, Ramp built a full-context background coding agent on Modal Sandboxes that generates code changes and writes them back into commits and pull requests. Modal's suitability for reinforcement learning is supported further by its RL infrastructure write-up, the Applied Compute RL case study, official GRPO implementations with TRL and verl, and customers such as Cognition that report using Modal for both RL infrastructure and production inference. For teams running RL rollouts with small open source models, Modal's combination of elastic GPU scaling, per-second billing without mandatory reservations, and integrated platform capabilities makes it a strong infrastructure choice for experimentation. Explore Modal's GRPO example to see implementation patterns for GRPO and other RL algorithms.
Explore Modal's GRPO example to see implementation patterns for GRPO and other RL algorithms.
Explore Modal's GRPO exampleModels from roughly 360M to 4B parameters reduce model-state memory and per-token compute relative to frontier models, which allows higher rollout parallelism on the same hardware budget. The exact number of model calls an RL run needs is configurable rather than fixed: PPO exposes total episodes, and GRPO exposes generations per prompt, generation batch size, and update count. Wall-clock speed still depends on sequence lengths, batching, kernels, and hardware utilization, not on parameter count alone.
Modal bills compute by the second with no minimum usage increment and does not require reservations, and Functions scale to zero by default. Warm pools, container buffers, and scaledown windows are sizing controls, so teams choose exactly how much standing capacity a workload keeps. Fast cold starts reduce the cost of bursty rollout patterns, and having training, inference, and Sandboxes on one platform reduces cross-service integration and operational overhead.
Qwen3 models offer dual reasoning modes whose benefit varies widely by benchmark and which cost additional generated tokens. SmolLM models emphasize training transparency, publishing pretraining data, recipes, code, and checkpoints, with SmolLM3's mid-training and post-training datasets still pending per its model card. Pick based on which tradeoff your project needs, and validate on your own task rather than on benchmark headlines.
Yes. All six models are released under Apache 2.0, which permits commercial fine-tuning subject to the license's notice, attribution, redistribution, trademark, and patent conditions. Modal supports single-GPU and multi-GPU fine-tuning; the maximum model size that fits on one GPU depends on the GPU and training method, and Modal's official examples include parameter-efficient fine-tuning of Qwen3-14B on a single GPU. Multi-node training is available for larger distributed workloads. For checkpoint management during long runs, use Modal Volumes; Memory Snapshots are a separate cold-start feature, not a checkpointing mechanism.
Modal provides compute isolation through gVisor containerization and virtualization, states that it has completed a SOC 2 Type II audit, and encrypts data in transit and at rest, per its security documentation. For RL applications executing untrusted agent code, Modal sandboxes add an isolated execution environment with configurable network controls, and connection tokens authenticate access to servers running inside a sandbox. Enterprise teams can execute a BAA for HIPAA-compliant workloads.