google/orbax

Issues

• Failed to get flag value for EXPERIMENTAL_ORBAX_USE_DISTRIBUTED_PROCESS_ID

  #1147 opened 4 months ago by ZaberKo
  1

• Github releases page not up to date with pypi releases

  #1121 opened 4 months ago by nouiz
  2

• Flax NNX checkpointing is not that straightforward

  #1105 opened 4 months ago by prabhudavidsheryl
  1

• Warnings When Restoring the Params

  #1110 opened 4 months ago by richardmkit
  1

• Can't checkpoint Numpy RNG state

  #1071 opened 4 months ago by rainx0r
  6

• NotImplementedError: CopyRange not supported

  #1048 opened 4 months ago by cool-RR
  3

• Multi-host Checkpointing Error

  #999 opened 6 months ago by ssenan
  12

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

  #1049 opened 5 months ago by galah92
  1

• TypeError: write(): incompatible function arguments.

  #1039 opened 5 months ago by FranzKnut
  1

• Type Handler registry collision warnings

  #1019 opened 5 months ago by varunagrawal
  0

• Restoring to CPU

  #986 opened 6 months ago by lachinov
  2

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

  #980 opened 6 months ago by jaro-sevcik
  1

• Saving using the refactored manager

  #941 opened 6 months ago by NITHISHM2410
  4

• How to restore on a CPU a checkpoint saved on a GPU?

  #646 opened 6 months ago by paulbarbier
  1

• How to restore a variable from checkpoint saved in cpu back in cpu when you have both gpu and cpu?

  #657 opened 6 months ago by PriyeshV
  5

• Struggling to restore metadata on other device

  #678 opened 6 months ago by thorben-frank
  6

• Installation error when installing T5x

  #719 opened 6 months ago by jntdst
  1

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

  #742 opened 6 months ago by qGentry
  3

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

  #803 opened 6 months ago by amifalk
  1

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

  #818 opened 6 months ago by ZaberKo
  1

• WARNING:absl:SaveArgs.aggregate is deprecated

  #839 opened 6 months ago by jiagaoxiang
  1

• Checkpointing issue in a distributed setup

  #844 opened 6 months ago by vladyorsh
  1

• Basic save/restore using OCDBT not working

  #924 opened 7 months ago by kvablack
  3

• Cannot restore sharded array on different machine

  #648 opened 7 months ago by kvablack
  8

• Tensorstore spec configuration

  #872 opened 8 months ago by minotru
  7

• CheckpointManager with a dataclass containing strings

  #825 opened 8 months ago by LeonEricsson
  6

• Help needed for restoring an orbax checkpoint

  #841 opened 8 months ago by jiagaoxiang
  0

• Custom strategy of keeping checkpoints

  #806 opened 8 months ago by Karina1997
  1

• New interface does not support `None` in pytrees

  #721 opened 9 months ago by PhilipVinc
  5

• New interface does not support empty dicts in pytrees

  #720 opened 9 months ago by PhilipVinc
  1

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

  #741 opened 10 months ago by allen-adastra
  10

• Orbax API migration questions

  #731 opened 10 months ago by haohuanw
  3

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

  #710 opened a year ago by qGentry
  4

• save_args_from_target alternatives?

  #722 opened 10 months ago by haohuanw
  4

• [Bug] Asyncio error while loading Flax weights

  #708 opened 10 months ago by mobley-trent
  7

• Tagging releases

  #709 opened a year ago by nhey
  2

• Top-level orbax import 0.5.* globally breaks logging

  #699 opened a year ago by qGentry
  3

• Error

  #667 opened a year ago by yimuu
  1

• Support checkpointing new-style `jax.random.key`

  #620 opened a year ago by hylkedonker
  6

• Parse structure of a saved PyTree checkpoint

  #683 opened a year ago by minotru
  1

• Strange behavior of saving sharded trainstate in GCP.

  #660 opened a year ago by chiamp
  3

• misstake submit

  #666 opened a year ago by hilanzy
  0

• Checkpoint Manager using different directory paths for save and restore

  #647 opened a year ago by svarunid
  2

• Importing orbax before TensorFlow silences important logging from TensorFlow

  #610 opened a year ago by kvablack
  0

• Array has been deleted

  #626 opened a year ago by fding
  4

• Async checkpointer failure handling

  #605 opened a year ago by hr0nix
  3

• Dependency array-record Not Available on ARM64 Architecture

  #607 opened a year ago by BeeGass
  2

• Issue loading checkpoint step during Paxml evaluation

  #575 opened a year ago by ashors1
  3

• AttributeError: 'Config' object has no attribute 'jax_coordination_service'

  #567 opened a year ago by EddieCunningham
  2

• Could you please provide the py.typed file for type checking?

  #561 opened a year ago by nasyxx
  0