Search results

Checkpoints do not preserve <code>AxisType.Explicit</code> meshes

Status: Open.

#2408 In google/orbax;

· jkyl opened on Sep 19, 2025

ImportError: cannot import name 'oex_orchestration_pb2' from 'orbax.export.protos'

type:bugSomething isn't workingSomething isn't working

Status: Open.

#2333 In google/orbax;

· kanglant opened on Sep 2, 2025

[MAINTAIN] orbax-checkpoint API does not expose user to Preservation Policy

checkpoint

type:others

Status: Open.

#2166 In google/orbax;

· shehper opened on Jul 24, 2025

Checkpointer memory leak

type:performance

Status: Open.

#2090 In google/orbax;

· sirfz opened on Jul 10, 2025

orbax uses logging.info directly and not the supplied log function, hence impossible to suppress logging w/o silencing the whole INFO log

checkpoint

type:bugSomething isn't workingSomething isn't working

Status: Open.

#1951 In google/orbax;

· qwer1304 opened on May 21, 2025

JaxRandKeySave and ArraySave give bogus warning about OCDBT

checkpoint

type:bugSomething isn't workingSomething isn't working

Status: Open.

#1950 In google/orbax;

· qwer1304 opened on May 21, 2025

Getting JaxRuntimeError key awaitable_signals_contract not found

better_errorsImprove the error reportingImprove the error reporting

checkpoint

Status: Open.

#1924 In google/orbax;

· bjenik opened on May 10, 2025

Restoring flax TrainState from two different checkpoint structures in the new API

checkpoint

type:supportFurther information is requestedFurther information is requested

Status: Open.

#1904 In google/orbax;

· cdelavegamartin opened on May 6, 2025

possible penzai pz.ParameterValue TypeHandler? Or, how to use orbax with penzai?

type:supportFurther information is requestedFurther information is requested

Status: Open.

#1833 In google/orbax;

· hrbigelow opened on Apr 15, 2025

Problems with orbax export documentation

documentationImprovements or additions to documentationImprovements or additions to documentation

export

Status: Open.

#1816 In google/orbax;

· reyammer opened on Apr 10, 2025

Incorrect / inconvenient type annotations inside orbax.checkpoint.checkpoint_args.register_with_handler

checkpoint

type:featureNew feature or requestNew feature or request

Status: Open.

#1718 In google/orbax;

· manulari opened on Mar 18, 2025

<code>step_prefix</code> cannot contain <code>_</code> -- Checkpoint manager does not recognized multiple <code>_</code>.

checkpoint

type:bugSomething isn't workingSomething isn't working

Status: Open.

#1499 In google/orbax;

· scott-yj-yang opened on Jan 14, 2025