Search results

Documentation for new <code>orb.checkpoint.v1</code> missing

documentationImprovements or additions to documentationImprovements or additions to documentation

Status: Open.

#2061 In google/orbax;

· PhilipVinc opened on Jun 30, 2025

Orbax doesn't work with GCS HNS bucket

type:performance

Status: Open.

#1955 In google/orbax;

· samos123 opened on May 22, 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

[RFC] orbax.checkpoint.v1 design

checkpoint

type:featureNew feature or requestNew feature or request

Status: Open.

Task

#1624 In google/orbax;

· orbax-dev opened on Feb 21, 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

<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