-
Notifications
You must be signed in to change notification settings - Fork 71
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Move read-time dtype casting into
ArrayReadSpec.
#2701
opened Dec 12, 2025 by
copybara-service
bot
Loading…
Fix the logic of oex.utils.get_lowering_platforms() as it didn't lower the passed native_serialization_platforms.
#2699
opened Dec 12, 2025 by
copybara-service
bot
Loading…
Refactor jax2obm configuration using a dataclass.
#2698
opened Dec 12, 2025 by
copybara-service
bot
Loading…
Refine bfloat16 conversion in Orbax Export utils.
#2697
opened Dec 12, 2025 by
copybara-service
bot
Loading…
#v1 Modify background thread execution to allow the runner to accept multiple concurrent tasks for the background event loop. This allows scheduling multiple operations, like async directory creation and background commits, on the same event loop.
#2686
opened Dec 10, 2025 by
copybara-service
bot
Loading…
Refactor ServingConfig to use ObmExportOptions dataclass.
#2675
opened Dec 8, 2025 by
copybara-service
bot
Loading…
Enforce topology check in StandardCheckpointHandler.restore
#2663
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Remove v0 logic out of
OrbaxLayout which now only represents v1 checkpoints/layouts
#2662
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Split v0 checkpoint format/layout logic out from OrbaxLayout and into seperate V0Layout
#2661
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Add testing to verify backwards compatibility when missing _CHECKPOINT_METADATA or saving pytree directly to checkpoint.
#2660
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Fix composite handler's extraction logic for handler type strings when missing _CHECKPOINT_METADATA
#2659
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Integrate TPU Inference Converter V2 for bfloat16 optimization in Orbax export.
#2658
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Ensure metadata and loading works correctly when missing _CHECKPOINT_METADATA file.
#2656
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Temporarily disable proto compilation for orbax/experimental/model in doc serving script.
#2644
opened Dec 2, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
no:milestone will show everything without a milestone.