Search results

How to force overwrite existing checkpoints?

checkpoint

type:supportFurther information is requestedFurther information is requested

Status: Open.

#2657 In google/orbax;

· salfaris opened on Dec 4, 2025

Enable saving (and loading) zero size arrays

type:featureNew feature or requestNew feature or request

Status: Open.

#2587 In google/orbax;

· pcmoritz opened on Nov 15, 2025

saving with default/global mesh is broken

type:bugSomething isn't workingSomething isn't working

Status: Open.

#2545 In google/orbax;

· PhilipVinc opened on Nov 5, 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

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

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

CheckpointManager randomly crashes the kernel of the JupyterLab

checkpoint

type:bugSomething isn't workingSomething isn't working

Status: Open.

#1408 In google/orbax;

· vfdev-5 opened on Nov 27, 2024

<code>ocp.tree.serialize_tree</code> filtering logic for sequences with empty leaves

checkpoint

type:bugSomething isn't workingSomething isn't working

Status: Open.

#1356 In google/orbax;

· JesseFarebro opened on Nov 19, 2024