Search results

orbax calls to <code>jax.lax.slice_in_dim</code> trigger compile storm

type:bugSomething isn't workingSomething isn't working

Status: Open.

#2934Â?In google/orbax;

· jfc4050 opened on Feb 24, 2026

<code>orbax-checkpoint</code> broken on Windows since 0.11.33

type:bugSomething isn't workingSomething isn't working

Status: Open.

#2914Â?In google/orbax;

· JeppeKlitgaard opened on Feb 19, 2026

Checkpoint save fails with ENOENT during OCDBT atomic rename on TPU v5litepod

type:bugSomething isn't workingSomething isn't working

Status: Open.

#2837Â?In google/orbax;

· jaisong123 opened on Feb 1, 2026

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