Feat: Activation Checkpointing - #154

Open

Chamberlain0w0 wants to merge 7 commits into

feat/activation_checkpointing

Commits on Sep 28, 2026

feat: init ac
Chamberlain0w0
committed
f0af008
refactor: cleanup & add comments
Chamberlain0w0
committed
fb3b526
fix: resolve comments
Chamberlain0w0
committed
830788b
fix: use autocast context saved in forward
Chamberlain0w0
committed
7da2044
fix: adapt to new zero flags
Chamberlain0w0
committed
1c383d0
fix: adapt to new test config
Chamberlain0w0
committed
a96ae32

Commits on Sep 29, 2026

feat: support Qwen3 recompute
Chamberlain0w0
committed
a6ff670