-
Notifications
You must be signed in to change notification settings - Fork 24
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.
#988
opened Jul 4, 2024 by
copybara-service
bot
Loading…
Migrate jax dependency to >=4.30 for jax.export API.
#987
opened Jul 3, 2024 by
copybara-service
bot
Loading…
Add structured logging for emergency checkpoint manager.
#981
opened Jun 29, 2024 by
copybara-service
bot
Loading…
Introduce replica_axis_index in init of emergency checkpoint manager
#968
opened Jun 21, 2024 by
copybara-service
bot
Loading…
Roll forward: Improve tensorstore I/O efficiency
#961
opened Jun 20, 2024 by
copybara-service
bot
Loading…
Add memory limiting functionality for save, in addition to restore.
#956
opened Jun 17, 2024 by
copybara-service
bot
Loading…
Orbax: Refactor tensorstore spec construction and support local temporary index path
#945
opened Jun 11, 2024 by
copybara-service
bot
Loading…
[JAX] Update users of jax.tree.map() to be more careful about how they handle Nones.
#938
opened Jun 9, 2024 by
copybara-service
bot
Loading…
*** DNS *** Demo code to fix eval job killed due to low duty cycle.
#922
opened Jun 4, 2024 by
copybara-service
bot
Loading…
End support for legacy Pax formats with directory structure:
#882
opened May 16, 2024 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Updated in the last three days: updated:>2024-07-02.