feat: add Energon-owned SFT sequence packing - #4070

Open

rohitrango wants to merge 12 commits into

NVIDIA-NeMo:mainfrom

rohitrango:rohit/sft_v2_stage2

Commits on Sep 10, 2026

feat: add Energon-owned SFT sequence packing
rohitrango
committed
ad2110b
fix: transport packed boundaries across TP ranks
rohitrango
committed
1277857
test: cover Energon-owned sequence packing
rohitrango
committed
2615ba9
chore: add packing module to pyrefly
rohitrango
committed
eb62214
chore: add test copyright headers
rohitrango
committed
da8a263
Update nemo_rl/data/packing/algorithms.py
rohitrango
authored
ac2519d
test(sft): add CP2 Energon packing recipe
rohitrango
committed
51cdc68
fix(sft): address Energon packing review feedback
rohitrango
committed
31d3111
docs(sft): clarify packed sample metrics
rohitrango
committed
1fa1033
feat(sft): allow all Energon packing algorithms
rohitrango
committed
295add7
Merge branch 'main' into rohit/sft_v2_stage2
rohitrango
authored
f8eaef7
fix(sft): satisfy lint checks
rohitrango
committed
619c3d6