Skip to content

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

Open
prishajain1 wants to merge 1 commit into
mainfrom
prisha/flux_training
Open

refactor Flux transformer to use scanned blocks, dynamic checkpointing, and decoupled projections#417
prishajain1 wants to merge 1 commit into
mainfrom
prisha/flux_training

Flux training: Implement scanned blocks, dynamic gradient checkpointi…

11ddfef
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jun 12, 2026 in 7s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

11ddfef Author: @prishajain1 <pri*****in​@google.com>

(Only the first commit for a unique contributor is listed.)