Dan Saunders 59cd472504 SP cu_seqlens fix, refactor (#2495) ...

* working on masking fix

* refactor and fix multipack seqlens

* pre-commit fix

* adding smoke test

* using existing packed seqlens util

* log warning re: logged losses / gradient scaling per rank

2025-04-07 14:47:57 -04:00

..

solo

Flex Attention + Packing with BlockMask support (#2363)

2025-04-05 18:02:57 -04:00

__init__.py

Attempt to run multigpu in PR CI for now to ensure it works (#1815) [skip ci]

2024-08-09 11:50:13 -04:00

test_eval.py

Updates for trl 0.16.0 - mostly for GRPO (#2437) [skip ci]

2025-03-31 15:47:11 -04:00

test_gemma3.py

gemma3 packing fixes (#2449)

2025-03-31 17:15:23 -04:00

test_llama.py

llama4 support (#2493)

2025-04-07 10:49:15 -04:00

test_qwen2.py

Update dependencies and show slow tests in CI (#2492)

2025-04-05 17:41:31 -04:00

test_ray.py

llama4 support (#2493)

2025-04-07 10:49:15 -04:00

test_sp.py

SP cu_seqlens fix, refactor (#2495)

2025-04-07 14:47:57 -04:00