- Notifications
You must be signed in to change notification settings - Fork61
Pull requests: google/orbax
Author
Uh oh!
There was an error while loading.Please reload this page.
Label
Uh oh!
There was an error while loading.Please reload this page.
Projects
Uh oh!
There was an error while loading.Please reload this page.
Milestones
Uh oh!
There was an error while loading.Please reload this page.
Reviews
Assignee
Assigned to nobodyLoading
Uh oh!
There was an error while loading.Please reload this page.
Sort
Pull requests list
Copy Bf16Optimizaion implementation to Orbax.
#2149 openedJul 22, 2025 bycopybara-servicebotLoading…
Export multiple stablehlos for a JAX poly function, based on discrete values for symbols provided by user
#2148 openedJul 22, 2025 bycopybara-servicebotLoading…
Small refactoring: Pass exactly what the function needs
#2146 openedJul 21, 2025 bycopybara-servicebotLoading…
Pass
file_options
toTemporaryPath
only infrom_final
and not increate
. #2145 openedJul 21, 2025 bycopybara-servicebotLoading…
Move blocking save implementation details from saving to saving_utils.
#2140 openedJul 18, 2025 bycopybara-servicebotLoading…
v1: Update LeafHandlerRegistry to work with handler_type only.
#2133 openedJul 18, 2025 bycopybara-servicebotLoading…
Move implementation details of Orbax V1
save_pytree
andsave_checkpointables
into helper file. #2127 openedJul 17, 2025 bycopybara-servicebotLoading…
Fixes for safetensors import error from previous rollback.
#2125 openedJul 17, 2025 bycopybara-servicebotLoading…
Add a new parameter
background_thread_count
to allow users to set the number of threads for ThreadedCheckpointDeleter. #2116 openedJul 16, 2025 bycopybara-servicebotLoading…
[REX] Enable ttl based deletion of step and substeps
#2114 openedJul 16, 2025 bycopybara-servicebotLoading…
[Orbax Model] Save XLA compile options per platform.
#2113 openedJul 15, 2025 bycopybara-servicebotLoading…
Add an unit test for oex_orchestration.create().
#2108 openedJul 14, 2025 bycopybara-servicebotLoading…
Use Checkpoint Manager to benchmark in Orbax V0.
#2102 openedJul 14, 2025 bycopybara-servicebotLoading…
Add branching logic to
loading.py
and refactor the checkpoint handler logic. #2094 openedJul 10, 2025 bycopybara-servicebotLoading…
Make the temp directory generation be unique across CheckpointManager when shared barrier_sync_key_prefix.
#2086 openedJul 9, 2025 bycopybara-servicebotLoading…
Add CheckpointLayout protocol for various checkpoint formats to inherit from.
#2081 openedJul 7, 2025 bycopybara-servicebotLoading…
ProTip! Follow long discussions withcomments:>50.