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

Measures generative AI inference performance with token-level metrics (TTFT, inter-token latency), latency, and throughput under realistic traffic patterns. Provides a multiprocess engine, real-time TUI dashboard, extensible plugins, and integrations for telemetry and result uploads, aimed at inference benchmarking and capacity planning.

GitHub
AI Train2019

Train and experiment with multi-billion to trillion-parameter transformer models on large GPU clusters using GPU-optimized building blocks and reference training scripts; offers advanced parallelism and mixed-precision support for research teams and ML engineers.

GitHub

Indexes full text of visited web pages and local files on a self‑hosted server so you can search your personal knowledge from a web UI, terminal, CLI, or an AI assistant. Runs without mandatory telemetry, offers a browser extension for automatic capture, and supports optional semantic search via a configurable embeddings endpoint.