maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

Published 20 hours ago • verified publisher icon google

Initialize jax distributed when checkpointing is enabled

Open jonb377 opened this issue 5 months ago • 4 comments

Nightly tests are failing due to jax.distributed not being initialized in the synchronous checkpointing case.

jonb377 avatar Sep 16 '24 23:09 jonb377