AIAny
AI Train2023
Icon for item

OpenRLHF

Trains LLMs with RLHF at scale by splitting actor, critic, reward, and reference models across separate GPU groups via Ray, with vLLM-accelerated generation and DeepSpeed ZeRO-3. Supports PPO, GRPO, REINFORCE++, DPO, plus async and agentic multi-turn RL.

Introduction

Most open RLHF stacks choke past a few billion parameters because actor, critic, reward, and reference models fight over the same GPUs. OpenRLHF's bet is scheduling, not just kernels: it uses Ray to place each model on its own GPU group, hands rollout generation to vLLM, and runs the rest on DeepSpeed ZeRO-3 — which is why 70B+ RLHF runs become routine instead of heroic.

What Sets It Apart
  • Disaggregated placement means you can right-size GPUs per role (a big actor, a small reward model) instead of co-locating everything and wasting memory.
  • vLLM-driven sampling removes the usual RLHF bottleneck, where slow generation, not gradient steps, dominates wall-clock time.
  • One algorithm-agnostic loop covers PPO, GRPO, RLOO, REINFORCE++/baseline, and DAPO, alongside SFT, reward-model training, and DPO/IPO — so switching methods is a flag, not a rewrite.
  • Async and agentic multi-turn modes via a token-in-token-out interface let it train reasoning and tool-use policies, not just single-turn preference tuning.
Who It's For

Great fit if you're running serious RLHF or RL-for-reasoning experiments on multi-node clusters and want vLLM throughput without gluing the pieces together yourself. Look elsewhere if you have a single GPU or just need quick DPO/SFT on small models — the Ray-plus-vLLM-plus-DeepSpeed setup is real operational overhead that only pays off at scale.

Information

  • Websitegithub.com
  • OrganizationsOpenRLHF Team, ByteDance, Tencent, Netease Fuxi AI Lab, Alibaba Group
  • AuthorsJian Hu, Xibin Wu, Zilin Zhu, Xianyu, Weixun Wang, Dehao Zhang, Yu Cao
  • Published date2023/07/30

More Items

GitHub
AI Infra2026

Compresses tool outputs, logs, files and RAG chunks locally before they reach the LLM to reduce token usage while caching originals for on-demand retrieval. Provides proxy/library/wrap modes, JSON- and code-aware compressors, and reversible compression.

GitHub
AI Infra2026

Provides a self‑hosted OpenAI‑compatible gateway that exposes one API endpoint to route requests to 40+ model providers, issue revocable virtual keys, enforce budgets before spend, and record usage.

GitHub
AI Coding2023

A statically typed, parallel programming language that prevents AI-introduced bugs by making correctness claims (laws) require machine-checked proofs. Compiles to native code and targets CPU/GPU parallelism with a fast proof checker.