#168 · Primary category: Deep Learning Frameworks
lmm-r1
Extend OpenRLHF to support LMM RL training for reproduction of DeepSeek-R1 on multimodal tasks.
Project last updated:05/14/25
GitHub Stars
848
Forks
53
Contributors
74
License
Apache-2.0
Why we included this project
OpenRLHF does not natively cover multimodal RL training, so teams wanting DeepSeek-R1-style rule-based RL on vision-language models usually have to assemble their own pipeline. LMM-R1 closes that gap by adapting OpenRLHF so 3B LMMs can be trained with a two-stage scheme, text-only reasoning first and multimodal generalization second, and ships the VerMulti dataset with training scripts to reproduce the reported geometry and perception-reasoning gains on local hardware. Much of the code has been upstreamed into OpenRLHF-M, so groups already on that stack can adopt the approach without maintaining a fork. For researchers and engineers applying rule-based RL to custom VLMs, it provides working data and configs, not just a paper.
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.