Commit e25f787

committed

use pytorch version only

1 parent bc068d8 commit e25f787Copy full SHA for e25f787

6 files changed

+747

-348

lines changed

src/liger_kernel
- chunked_loss
  - fused_linear_ppo.py
- ops
  - __init__.py
  - grpo_loss.py
- transformers
  - grpo_loss.py
test
- chunked_loss
  - test_grpo_loss.py
- transformers
  - test_grpo_loss.py

6 files changed

+747

-348

lines changed

Comments

(0)