issues
search
google
/
orbax
Orbax provides common checkpointing and persistence utilities for JAX users
https://orbax.readthedocs.io/
Apache License 2.0
304
stars
36
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
I have trouble to restore the same checkpoint twice.
#1300
copybara-service[bot]
closed
2 weeks ago
0
Re-expose Metadata symbol.
#1299
copybara-service[bot]
closed
2 weeks ago
0
Refactoring: move type_handlers.py to _src/serialization
#1298
copybara-service[bot]
closed
2 weeks ago
0
Add notes to Barrier error `XlaRuntimeError(DEADLINE_EXCEEDED)` with actionable info.
#1297
copybara-service[bot]
closed
2 weeks ago
0
Make `step.step_metadata_of_checkpoint_path()` concurrent to improve performance.
#1296
copybara-service[bot]
closed
1 week ago
0
Move Orbax `path` package under `_src`.
#1295
copybara-service[bot]
closed
2 weeks ago
0
AttributeError: module 'orbax.checkpoint.metadata' has no attribute 'Metadata'
#1294
hanzhi713
opened
3 weeks ago
2
Internal change.
#1293
copybara-service[bot]
closed
3 weeks ago
0
Internal change.
#1292
copybara-service[bot]
closed
1 week ago
0
Make `NameFormat.find_all` impls concurrent.
#1291
copybara-service[bot]
closed
3 weeks ago
0
Add user-facing Root/StepMetadata classes and internal serialization modules.
#1290
copybara-service[bot]
closed
1 week ago
0
Minimize usage of `step.checkpoint_steps()` as it is not compatible with all types of finalized checkpoints.
#1289
copybara-service[bot]
closed
3 weeks ago
0
StandardSave fails with confusing error when given scalars or arrays
#1288
garymm
closed
3 weeks ago
3
Internal change.
#1287
copybara-service[bot]
closed
3 weeks ago
0
Fix a bug in Orbax checkpointing where None values PyTree are not handled correctly. This is caused by a recent update in jax where None values are no longer considered as a leaf node: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-34-october-4-2024.
#1286
copybara-service[bot]
opened
3 weeks ago
1
Print the model_param shape instead of the model_params.
#1285
copybara-service[bot]
closed
3 weeks ago
0
Move the Composite mapping to a separate file.
#1284
copybara-service[bot]
closed
3 weeks ago
0
Fix serialization case in which `global_shape` instead of `shard_shape` would be chosen as the TensorStore chunk shape in case of degenerate fragments (no primary replica shards on a host).
#1283
copybara-service[bot]
closed
3 weeks ago
0
Internal change.
#1282
copybara-service[bot]
opened
3 weeks ago
0
Internal change.
#1281
copybara-service[bot]
closed
3 weeks ago
0
Internal change.
#1280
copybara-service[bot]
closed
3 weeks ago
0
update link to docs under checkpoint_manager.py.
#1279
copybara-service[bot]
closed
3 weeks ago
0
Internal change.
#1278
copybara-service[bot]
closed
3 weeks ago
0
Increase version to 0.8.0.
#1277
copybara-service[bot]
closed
3 weeks ago
0
Add user-facing `TreeMetadata` object returned by `PyTreeCheckpointHandler.metadata`. This object mimics an ordinary PyTree to make the change unnoticeable to most users, but also has additional accessible properties not included in any tree mapping operations.
#1276
copybara-service[bot]
opened
4 weeks ago
0
Rename `TreeMetadata` to `InternalTreeMetadata`.
#1275
copybara-service[bot]
closed
4 weeks ago
0
Wires up `ExportManager` to Orbax Model.
#1274
copybara-service[bot]
closed
4 weeks ago
0
Adds support for TF pre/post-processing to `ObmModule` and `ObmExport`.
#1273
copybara-service[bot]
closed
4 weeks ago
0
Move `serialization` to `_src`
#1272
copybara-service[bot]
closed
2 weeks ago
0
Move `tree` package to `_src`.
#1271
copybara-service[bot]
closed
3 weeks ago
0
Move `metadata` package to `_src`. Continue exposing commonly-used symbols.
#1270
copybara-service[bot]
closed
4 weeks ago
0
Internal change.
#1269
copybara-service[bot]
closed
1 month ago
0
Ensure that temporary directories are only finalized once.
#1268
copybara-service[bot]
closed
1 month ago
1
modify snapshot folder construction to not rely on the original step path.
#1267
copybara-service[bot]
closed
1 month ago
0
always clean up any existing local temporary directories on emergency checkpoint manager creation
#1266
copybara-service[bot]
closed
1 month ago
0
Use 3.10 type annotations for code under `_src/arrays`.
#1265
copybara-service[bot]
closed
1 month ago
0
change default value for `cleanup_tmp_directories` to True for emergency checkpoint manager
#1264
copybara-service[bot]
closed
1 month ago
0
Add shutdown logging wrapper.
#1263
copybara-service[bot]
opened
1 month ago
0
rollback snapshot change.
#1262
copybara-service[bot]
closed
1 month ago
0
Improve handlers/args documentation. Also adjust some public symbols to avoid duplicate imports (and resulting breakage) and to get docs to render better.
#1261
copybara-service[bot]
closed
1 month ago
0
Update fix:
#1260
lamcodeofpwnosec
opened
1 month ago
1
Move `logging` to `_src`.
#1259
copybara-service[bot]
closed
1 month ago
0
Add announcement for `strict` option.
#1258
copybara-service[bot]
closed
1 month ago
0
Rollback of `strict` option.
#1257
copybara-service[bot]
closed
1 month ago
0
Rollback of `strict` option.
#1256
copybara-service[bot]
closed
1 month ago
0
Small change to fix test breakage
#1255
copybara-service[bot]
closed
1 month ago
1
Adds the ability to specify a checkpoint path to the Orbax export path.
#1254
copybara-service[bot]
closed
1 month ago
1
Fix user `logging.basicConfig` issue.
#1253
copybara-service[bot]
closed
1 month ago
0
Add `should_save_fn` for emergency CheckpointManager.
#1252
copybara-service[bot]
closed
1 month ago
0
De-duplicate `get_ts_context` usages and move to ts_utils.
#1251
copybara-service[bot]
closed
1 month ago
0
Previous
Next