alibaba / alibaba/ROLL

ROLL 是否可以加速reward模型(非LLM)计算与LLM GRPO的高效协同训练

Open
#230 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
3.4k
Forks
312
Avg merge
1h 2m
Merged PRs (30d)
2

Description

我有个场景是用Diffusion模型作为reward,导致reward计算相对较慢,且占用额外的显卡。请问是否有方式可以加速这类的reward的计算,因为在计算reward时,训练GRPO的显卡处于等待中。

我已经做的是对Diffusion模型本身和Diffusion去噪做了加速,也做成了一个reward server、异步请求reward计算等。

Contributor guide

No contributing guide indexed for this repository

Research direction

No files, tests, or entry points are named. Start by tracing the Diffusion reward calculation and the GRPO training synchronization, then review the existing reward-server and asynchronous-request approach described in the issue. Done means a documented, agreed design that reduces reward-related waiting and coordinates GPU use efficiently.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
ai, distributed-systems, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.