orbax

orbax copied to clipboard

Handle the special case of `_get_concrete_function` when f is already a `PolymorphicFunction`.

1

Handle the special case of `_get_concrete_function` when f is already a `PolymorphicFunction`. This will allow the name specified in `TensorSpec` to propagate to `named_signature`.

copybara-service[bot]

add the `servo signature` to the `Pipeline`.

1

add the `servo signature` to the `Pipeline`.

copybara-service[bot]

Getting JaxRuntimeError key awaitable_signals_contract not found

1

Over the last few days using orbax checkpoint installed from the repo directly I've had increasing numbers of grpc key awaitable_signals_contract not found issues pop up. This is on midsize...

bjenik

Use the new preservation policy in CheckpointManager.

copybara-service[bot]

Re-organize OCP v0 and v1 docs directory structure.

copybara-service[bot]