perf: selective activation checkpointing feature support - #2280

Merged

terrykong merged 20 commits into

mainfrom

sj/selective-recompute

May 22, 2026

Commits on Apr 17, 2026

Add selective activation recompute support
seonjinn
committed
6621d76

Commits on Apr 20, 2026

Update nemo_rl/models/policy/__init__.py

seonjinn
and
terrykong
authored
2c20d69
Update nemo_rl/models/megatron/setup.py

seonjinn
and
terrykong
authored
391eb9b
Update examples/configs/grpo_math_1B.yaml

seonjinn
and
terrykong
authored
38f92e5
Update nemo_rl/models/megatron/setup.py

seonjinn
and
terrykong
authored
e5acdd8

Commits on Apr 23, 2026

Merge branch 'main' into sj/selective-recompute
seonjinn
authored
72d1d15

Commits on Apr 29, 2026

Merge branch 'main' into sj/selective-recompute
seonjinn
authored
79486ec

Commits on May 14, 2026

Merge remote-tracking branch 'origin/main' into sj/selective-recompute
seonjinn
committed
8102a0d

Commits on May 16, 2026

test: add unit tests for recompute_granularity dispatch
seonjinn
committed
810260c

Commits on May 19, 2026

Merge main into sj/selective-recompute
seonjinn
committed
c40a482

Commits on May 20, 2026

Merge branch 'main' into sj/selective-recompute
seonjinn
committed
05ca762
Merge branch 'main' into sj/selective-recompute
seonjinn
authored
638d22a

Commits on May 22, 2026