orbax

orbax copied to clipboard

Internal changes to support more export / load options.

1

Internal changes to support more export / load options.

copybara-service[bot]

Internal change.

copybara-service[bot]

Internal change.

copybara-service[bot]

Internal change.

copybara-service[bot]

Internal change.

copybara-service[bot]

Fix error checking in async portion of CheckpointManager save.

copybara-service[bot]

Refactor training time saved measurement so that all non-primary hosts report it as well.

1

Refactor training time saved measurement so that all non-primary hosts report it as well.

copybara-service[bot]

Incompatibility with Haiku

5

Reopening an issue regarding incompatibility with Haiku naming conventions (similar to [previous issue](https://github.com/google/orbax/issues/339)). This is not problematic in v0.3.5 Sample code: ```python from jax import numpy as jnp import orbax.checkpoint...

chriscarmona

Internal change.

1

Internal change.

copybara-service[bot]

It would be great if you could provide the empty py.typed file for type checking. I'm a little confused about the structure of two packages in one repo, and not...

nasyxx