#145 · Primary category: Deep Learning Frameworks
safe-rlhf
Safe RLHF: Constrained Value Alignment via Safe Reinforcement Learning from Human Feedback
Project last updated:11/24/25
GitHub Stars
1.6K
Forks
134
Contributors
4
License
Apache-2.0
Why we included this project
Beaver is an RLHF training framework from the PKU-Alignment group at Peking University, built for researchers who want to treat safety as an explicit constraint rather than a post-training fix. It covers the usual alignment stages, from supervised fine-tuning through reward and cost model training to standard RLHF and the constrained Safe RLHF variant, and it runs on LLaMA, OPT, and Baichuan. The project also ships a large human-labeled preference dataset, up to about a million pairs scored for both helpfulness and harmfulness, along with pre-trained Beaver-7B checkpoints, so you can reproduce or extend published results without building the pipeline from scratch. Safety evaluation hooks for BIG-bench and GPT-4 help you verify that constrained training actually behaves as intended. If your team already uses DeepSpeed, the backend will feel familiar, though this is aimed at alignment research and custom preference datasets rather than one-click production deployment.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.