Skip to content

Commit 5ef76af

Browse files
authored
Add GRPO loss example (#1063)
1 parent c864ed4 commit 5ef76af

File tree

4 files changed

+1245
-1
lines changed

4 files changed

+1245
-1
lines changed

examples/README.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ Other Operations
5050
- :doc:`embedding.py <embedding>`: Embedding lookup operation
5151
- :doc:`all_gather_matmul.py <all_gather_matmul>`: All-gather operation followed by matrix multiplication
5252
- :doc:`all_reduce.py <all_reduce>`: All-reduce operation (one-shot)
53+
- :doc:`grpo_loss.py <grpo_loss>`: Group Relative Policy Optimization (GRPO) loss function
5354

5455
.. toctree::
5556
:maxdepth: 2

0 commit comments

Comments
 (0)