Make sure single JAX.random.key can be save by PyTreeCheckpointHandler#3097
Open
copybara-service[bot] wants to merge 1 commit intomainfrom
Open
Make sure single JAX.random.key can be save by PyTreeCheckpointHandler#3097copybara-service[bot] wants to merge 1 commit intomainfrom
copybara-service[bot] wants to merge 1 commit intomainfrom
Commits
Commits on Apr 15, 2026
- authored andcommitted
