-
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
Add directory to step statistics and integrate custom logger with emergency CM.
#1006
opened Jul 16, 2024 by
copybara-service
bot
Loading…
Move all work for
_write_metadata_file
into a background thread to avoid building metadata in the main thread. This is not all that costly, but it is O(n) where n is the number of arrays in the tree, so it can start to add up for trees with a lot of parameters.
#1001
opened Jul 12, 2024 by
copybara-service
bot
Loading…
Fix possible race condition in orbax root directory creation.
#1000
opened Jul 11, 2024 by
copybara-service
bot
Loading…
Ensure D2H copies are parallelized, following https://github.com/google/jax/pull/22169.
#997
opened Jul 10, 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…
Provide the utility functions orbax.export ApplyFn_Map to and from disk.
#983
opened Jul 2, 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…
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…
Previous Next
ProTip!
Filter pull requests by the default branch with base:main.