refactor Flux transformer to use scanned blocks, dynamic checkpointing, and decoupled projections#417

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxdiffusion:mainfrom

prisha/flux_trainingAI-Hypercomputer/maxdiffusion:prisha/flux_trainingCopy head branch name to clipboard