Orbax provides common checkpointing and persistence utilities for JAX users
304
stars
36
forks
source link
Add vlogs for emergency checkpointing so we can easily tell which indices of the array need to be loaded. #1364
Closed
copybara-service[bot] closed 5 days ago
Add vlogs for emergency checkpointing so we can easily tell which indices of the array need to be loaded.