CommonLoopUtils

CommonLoopUtils copied to clipboard

Implement recent changes to the DatasetIterator interface.

copybara-service[bot]

Implement __repr__() for several classes.

copybara-service[bot]

Add DataIterator and checkpoint handler for Grain.

copybara-service[bot]

This fix adjusts import to avoid accidentally pulling in keras in clu. https://github.com/google/CommonLoopUtils/pull/336

1

This fix adjusts import to avoid accidentally pulling in keras in clu. https://github.com/google/CommonLoopUtils/pull/336

copybara-service[bot]

Change deprecated `jax.tree_map` to avoid warnings:

1

Change deprecated `jax.tree_map` to avoid warnings: ``` DeprecationWarning: jax.tree_map is deprecated: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version). ```

copybara-service[bot]

Support sharding for get_parameter_overview

copybara-service[bot]

do not pull in tensorboard's keras callback

1

i am moving some of my models to keras3 and using jax backend for training. i am on a slightly older tensorboard (2.12) so when it pulls in keras callback...

haohuanw

Disable typeguard for TF

copybara-service[bot]

Fix mock_modules for `import x.y.z as a`

copybara-service[bot]

Fix type annotations to pass pytype checks.

copybara-service[bot]