Training
Reinforcement learning from human feedback (RLHF) with Proximal Policy Optimization (PPO) was central to early large language model alignment systems such as InstructGPT. PPO-based RLHF coordinates policy, reference, reward, and value models, making model selection and infrastructure capacity important parts of the training design. This guide covers Modal's training infrastructure and six open-weight checkpoints for PPO-based RLHF in 2026.

Modal provides serverless infrastructure for AI training, inference, batch processing, notebooks, and sandboxed execution. For PPO-based RLHF, teams can assign different GPU types and scaling policies to rollout, reward, value, and update workers instead of maintaining one fixed cluster for every phase.
PPO workloads alternate between generation and optimization, so utilization changes during a run. Modal lets teams define these resources in Python and provision them on demand:
OLMo-3.1-32B-Instruct-SFT is an SFT-stage checkpoint from Ai2's OLMo 3.1 pipeline. Ai2 publishes later preference-optimized and RLVR artifacts separately, making the SFT model useful when researchers want to apply and measure their own PPO objective.
The BF16 weights alone require roughly 64GB before gradients, activations, optimizer state, or the other PPO models are considered. Full-parameter PPO therefore needs sharding, offload, model colocation, or a distributed topology. LoRA can reduce trainable optimizer state but does not remove rollout, reference, reward, and value-model memory. Best For: Research teams that prioritize transparency and can support distributed 32B training.
| Stage | Checkpoint | Purpose |
|---|---|---|
| Supervised fine-tuning | OLMo-3.1-32B-Instruct-SFT | Instruction-following initialization before preference optimization |
| Preference optimization | OLMo-3.1-32B-Instruct-DPO | Optimization on preference pairs |
| Reinforcement learning | OLMo-3.1-32B-Instruct | Final checkpoint after the published RLVR stage |
OLMo-2-0325-32B-SFT is the supervised-fine-tuning checkpoint from Ai2's Tulu 3 sequence. The published downstream recipe used DPO and then GRPO rather than PPO, but the clean SFT stage remains valuable for teams comparing policy-gradient algorithms from the same initialization.
The short context window is a constraint for document, repository, and long-horizon agent tasks. It can also be an advantage for bounded preference datasets where predictable rollout lengths matter more than context capacity. Best For: Alignment researchers reproducing or comparing post-training stages on a well-documented checkpoint.
Qwen3-8B occupies a practical middle ground: it is large enough for meaningful reasoning and instruction-following experiments but small enough to replicate across the policy, reference, value, and reward topology with fewer GPUs than a 32B model.
Qwen chat templates expose thinking behavior that must be handled consistently across rollout generation, reference log-probabilities, and reward scoring. Fix the template and reasoning mode before collecting PPO data so formatting changes do not become an unintended reward signal. Best For: Teams that want a balanced, reproducible PPO baseline with broad open-source tooling support.
Qwen3-4B is a compact dense model suited to lower-cost PPO experimentation. Its size reduces policy-update and rollout costs relative to the larger checkpoints in this guide.
Long context is not free during PPO. KV cache, activations, and rollout storage grow with sequence length, so cap training trajectories based on the task rather than the model's advertised maximum. Best For: PPO prototyping, hyperparameter sweeps, and teams optimizing for training throughput.
SmolLM3-3B is the smallest checkpoint in this guide. Its 3B-parameter scale makes it suitable for PPO experiments where iteration speed and resource efficiency are the primary constraints.
SmolLM3-3B lowers the cost of rollout generation and makes it easier to test reward-model design, KL control, clipping ranges, and advantage estimation before scaling to a larger policy. Best For: Resource-constrained teams and researchers prioritizing rapid PPO iteration.
OLMo-2-1124-7B-SFT is a supervised-fine-tuning checkpoint in Ai2's OLMo 2 family. It provides a smaller OLMo starting point for teams that want transparent training artifacts without the resource requirements of a 32B policy.
The 7B checkpoint offers a middle ground between SmolLM3-3B and the 32B OLMo models. Its short context window keeps rollout memory bounded but limits long-document and long-horizon tasks. Best For: Teams seeking OLMo transparency with lower PPO infrastructure requirements.
PPO-based RLHF can coordinate policy, reference, reward, and value models alongside rollout workers. Modal lets teams assign separate Functions and resources to these roles while keeping orchestration in one codebase.
Modal provisions compute on demand and scales applicable workloads to zero when idle. Teams can run distributed training without maintaining a permanently allocated GPU cluster between experiments.
Modal runs user-defined containers, so teams can install PPO-capable frameworks such as OpenRLHF and Hugging Face TRL, along with the inference and distributed-training dependencies required by a checkpoint.
Modal's Python SDK defines images, Functions, GPU resources, storage, secrets, and scaling behavior in code rather than separate cluster configuration files.
The models in this guide range from 3B to 32B parameters. Modal's GPU infrastructure lets teams select hardware for each model size and scale from single-GPU experiments to multi-node training.
Run PPO-based RLHF workloads on elastic GPU infrastructure without managing a fixed cluster.
Explore Modal TrainingPPO-based RLHF uses a reward model trained from human preferences to score policy outputs, a value model to estimate returns, and a reference model to constrain policy drift. PPO was central to early aligned LLM systems because it provided a stable way to optimize language-model behavior against learned human preferences.
SmolLM3-3B has the smallest parameter count in this guide and therefore the lowest baseline memory requirement. Qwen3-4B is another practical option when teams want a compact policy with the Qwen ecosystem.
Modal provides on-demand GPU compute, multi-node training, persistent storage, custom container images, and code-first orchestration. Teams bring their chosen RL framework and can assign separate resources to policy, reference, reward, value, and rollout workloads.
SFT checkpoints have completed supervised fine-tuning but precede later preference-optimization or reinforcement-learning stages. Fully post-trained models have completed the vendor's broader alignment pipeline. Starting from an SFT checkpoint lets a team apply and evaluate its own PPO stage.
Requirements depend on model size, precision, optimizer, sequence length, batch size, sharding, offload, and worker placement. The 3B and 4B models are the most accessible for constrained setups, while full-parameter PPO with a dense 32B policy generally requires distributed high-memory GPUs.