#179 · Primary category: Deep Learning Frameworks
RL4LMs
A modular RL library to fine-tune language models to human preferences
Project last updated:03/01/24
GitHub Stars
2.4K
Forks
201
Contributors
8
License
Apache-2.0
Why we included this project
RL4LMs provides the building blocks for reinforcement-learning fine-tuning of language models, so you can focus on the reward and policy rather than the loop plumbing. You pick an on-policy algorithm like PPO or NLPO, a reward function from the library's broad set of NLG metrics, and a causal or seq2seq model as the policy. Predefined YAML configs make runs reproducible, and the library has been benchmarked across tasks like summarization, translation, dialogue, and table-to-text generation. It's a solid starting point for RLHF-style experiments or custom reward design, though the code is research-oriented and last updated in early 2024, so expect to adapt it for production.
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.