Skip to content

Gemma4e2b jax.checkpoint & remat fix - #4847

Open
mattdonati wants to merge 2 commits into
AI-Hypercomputer:mainfrom
mattdonati:gemma4e2b_checkpoint_fix
Open

Gemma4e2b jax.checkpoint & remat fix#4847
mattdonati wants to merge 2 commits into
AI-Hypercomputer:mainfrom
mattdonati:gemma4e2b_checkpoint_fix

add checkpoint boundaries to gemma4e2b

6799293
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Aug 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:

6799293 Author: @mattdonati <mat******ti​@gmail.com>

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