google/orbax

Issues

• Custom TypeHandler and "No per-process OCDBT checkpoint subdirs" warning

  #1326 opened 3 months ago by PhilipVinc
  3

• `No module named 'orbax.checkpoint._src.path'` when used as bazel pip dependency

  #1429 opened 2 months ago by hartikainen
  6

• `step_prefix` cannot contain `_` -- Checkpoint manager does not recognized multiple `_`.

  #1499 opened a month ago by scott-yj-yang
  2

• Checkpoint state invalid for both Zarr and OCDBT formats

  #1484 opened a month ago by plra
  1

• Weights of Restored model after checkpointing do not give same loss as model before saving

  #1402 opened 3 months ago by delara38
  1

• `ocp.tree.serialize_tree` filtering logic for sequences with empty leaves

  #1356 opened 3 months ago by JesseFarebro
  1

• _validate_params fails on zero-sized arrays

  #1309 opened 3 months ago by hrbigelow
  3

• Code duplication around tensorstore_spec logic in orbax-checkpoint

  #1241 opened 4 months ago by minotru
  2

• Warnings When Restoring the Params

  #1110 opened 6 months ago by richardmkit
  1

• Flax NNX checkpointing is not that straightforward

  #1105 opened 6 months ago by prabhudavidsheryl
  8

• TypeError: write(): incompatible function arguments.

  #1039 opened 6 months ago by FranzKnut
  2

• Type Handler registry collision warnings

  #1019 opened 7 months ago by varunagrawal
  0

• Multi-host Checkpointing Error

  #999 opened 7 months ago by ssenan
  12

• Tensorstore spec configuration

  #872 opened 9 months ago by minotru
  7

• Make GCS-style checkpointing configurable instead deciding it on is_gcs_path function output

  #710 opened a year ago by qGentry
  4

• `orbax/checkpoint/experimental/emergency` tests failing with `absl.flags._exceptions.DuplicateFlagError: The flag 'num_processes' is defined twice.`

  #1580 opened 10 days ago by GaetanLepage
  6

• `CheckpointManager.restore`: `directory` and `step=None` should not both be supplied.

  #1601 opened 5 days ago by jkyl
  2

• Can't checkpoint Numpy RNG state

  #1071 opened 9 days ago by rainx0r
  7

• CheckpointManager randomly crashes the kernel of the JupyterLab

  #1408 opened 19 days ago by vfdev-5
  7

• Latest orbax-export release is incompatible with latest orbax-checkpoint release

  #1314 opened a month ago by jakevdp
  1

• orbax-checkpoint 0.10.3 breaks grain

  #1456 opened a month ago by jakevdp
  1

• AttributeError: module 'orbax.checkpoint.metadata' has no attribute 'Metadata'

  #1294 opened a month ago by hanzhi713
  3

• Orbax Checkpointing and Flax.NNX require hacking to work together

  #1337 opened 2 months ago by hdrwilkinson
  2

• Restoring flax model checkpoints using orbax throws ValueError

  #1305 opened 2 months ago by ybangaru
  3

• StandardSave fails with confusing error when given scalars or arrays

  #1288 opened 3 months ago by garymm
  3

• Orbax import breaks logging. Again.

  #1248 opened 4 months ago by qGentry
  7

• Can't save model on Colab but can on Kaggle

  #1229 opened 4 months ago by windmaple
  1

• Hi, I am trying to save checkpoints using the following code:

  #1225 opened 4 months ago by raresdolga
  3

• Failed to get flag value for EXPERIMENTAL_ORBAX_USE_DISTRIBUTED_PROCESS_ID

  #1147 opened 5 months ago by ZaberKo
  2

• Github releases page not up to date with pypi releases

  #1121 opened 5 months ago by nouiz
  2

• NotImplementedError: CopyRange not supported

  #1048 opened 6 months ago by cool-RR
  3

• `TypeError: write(): incompatible function arguments`

  #1049 opened 6 months ago by galah92
  1

• Restoring to CPU

  #986 opened 7 months ago by lachinov
  2

• "SaveArgs.aggregate is deprecated" warning for empty tuples and empty NamedTuples

  #980 opened 7 months ago by jaro-sevcik
  1

• Saving using the refactored manager

  #941 opened 8 months ago by NITHISHM2410
  4

• Installation error when installing T5x

  #719 opened 8 months ago by jntdst
  1

• About using CPU backend as mock and unifying using multihost_utils wrappers along repo

  #742 opened 8 months ago by qGentry
  3

• Orbax cannot save numpy array with dtype=np.object_

  #803 opened 8 months ago by amifalk
  1

• New interface does not support custom empty pytree class inherited from dict

  #818 opened 8 months ago by ZaberKo
  1

• WARNING:absl:SaveArgs.aggregate is deprecated

  #839 opened 8 months ago by jiagaoxiang
  1

• Checkpointing issue in a distributed setup

  #844 opened 8 months ago by vladyorsh
  1

• Basic save/restore using OCDBT not working

  #924 opened 8 months ago by kvablack
  3

• CheckpointManager with a dataclass containing strings

  #825 opened 10 months ago by LeonEricsson
  6

• Help needed for restoring an orbax checkpoint

  #841 opened 10 months ago by jiagaoxiang
  0

• Custom strategy of keeping checkpoints

  #806 opened 10 months ago by Karina1997
  1

• New interface does not support `None` in pytrees

  #721 opened a year ago by PhilipVinc
  5

• New interface does not support empty dicts in pytrees

  #720 opened a year ago by PhilipVinc
  1

• Saving doesnt work and results in extra *.npy extension?

  #741 opened a year ago by allen-adastra
  10

• Orbax API migration questions

  #731 opened a year ago by haohuanw
  3

• save_args_from_target alternatives?

  #722 opened a year ago by haohuanw
  4