Sung Ching Liu a8f38c367c Flex Attention + Packing with BlockMask support (#2363)

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

..

__init__.py

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

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

test_flex.py

Flex Attention + Packing with BlockMask support (#2363)

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

test_grpo.py

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

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