[SAC] Refactor activation checkpointing to use centralized policy-based approach#2357

mori360 wants to merge 5 commits intopytorch:mainpytorch/torchtitan:mainfrom

mori360:acmori360/torchtitan:acCopy head branch name to clipboard