feat: add batch size scheduler for dynamic batch size warmup/rampup#39

xrsrke wants to merge 1 commit intodev-updated-againNousResearch/torchtitan:dev-updated-againfrom

phuc/add_batch_size_schedulerNousResearch/torchtitan:phuc/add_batch_size_schedulerCopy head branch name to clipboard