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 Infra2025

Defines a vendor-neutral JSON/YAML semantic model specification and tooling to exchange metrics, dimensions, lineage and other business semantics across analytics, AI and BI platforms; includes a core spec, validators, converters (dbt, GoodData, Salesforce) and example models.

GitHub
AI Train2025

An asynchronous, high-throughput framework for large-scale reinforcement learning and agentic training that scales to 1T+ MoE models and 1000+ GPUs, with native verifiers integration, end-to-end SFT/RL/evals, and Slurm/Kubernetes deployment; requires NVIDIA GPUs.

GitHub

Runs a self-hosted meeting bot and transcription API that joins Google Meet, Teams and Zoom and streams speaker-attributed transcripts in real time. Compiles meetings into a git-backed Markdown workspace and runs sandboxed agents on your infrastructure; Apache-2.0 and air-gap capable.