google/CommonLoopUtils issues and pull requests

#365 - Update clu to use inspect.get_annotations(cls) oveer cls.annotations

Pull Request - State: open - Opened by copybara-service[bot] 17 days ago - 1 comment

#364 - Change the count type from int32 to float32 to avoid overflows.

Pull Request - State: open - Opened by copybara-service[bot] 17 days ago - 1 comment

#363 - Fix when variable is None for include_stats is True or False

Pull Request - State: closed - Opened by copybara-service[bot] 24 days ago - 1 comment

#362 - Creating a NNX model with no bias results in a None being passed to the overview.

Pull Request - State: closed - Opened by copybara-service[bot] 25 days ago - 1 comment

#361 - Fix handling of empty parameters in parameter_overview.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#360 - jax.numpy.clip: update use of deprecated arguments.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 month ago

#359 - Update references to JAX's GitHub repo

Pull Request - State: open - Opened by copybara-service[bot] 3 months ago

#358 - lazy import tensorflow

Pull Request - State: open - Opened by haohuanw 3 months ago - 1 comment

#357 - update jnp.clip args

Pull Request - State: open - Opened by garymm 3 months ago

#356 - Adds `reshuffle_each_iteration` argument to `deterministic_data.create_dataset()`.

Pull Request - State: open - Opened by copybara-service[bot] 7 months ago

#355 - Allow running profiler on specific steps.

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago - 1 comment

#354 - Rollback add point clouds summary writer to tensorboard interface and metric writer.

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago - 1 comment

#353 - Add point clouds summary writer to tensorboard interface and metric writer.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago - 1 comment

#352 - Replace deprecated `jax.tree_` functions with `jax.tree.`

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#351 - Update GitHub Actions to checkout@v4 and setup-python@v5

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#350 - Fix bug in preprocess_spec.py: Remove `list()` that fails if tensor rank is unknown.

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#349 - Make g3pdb lazy again

Pull Request - State: open - Opened by copybara-service[bot] 10 months ago

#348 - Make sklearn import lazy

Pull Request - State: open - Opened by copybara-service[bot] 11 months ago

#347 - Rename dpsgd to orbax.contrib END_PUBLIC

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#346 - Rename DifferentiallyPrivateAggregateState to `orbax.contrib`

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#345 - Rename differentially_private_aggregate -> optax.contrib

Pull Request - State: open - Opened by copybara-service[bot] 11 months ago

#344 - Speed up clu.metric_writers by making rarely used imports lazy

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#343 - Release v0.0.12.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#342 - Updated `jax.tree_map` to `jax.tree_util.tree_map`. As of [JAX 0.4.26](https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-26-april-3-2024), `jax.tree_map` is deprecated. We should switch to `jax.tree.map` if/when the JAX minimum version is >=0.4.26 for CLU.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

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

Pull Request - State: open - Opened by copybara-service[bot] 11 months ago - 1 comment

#340 - Change deprecated `jax.tree_map` to avoid warnings:

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago - 1 comment

#339 - Add include_states="sharding" in the parameter overview to add a column to display the sharding without computing the mean and std.

Pull Request - State: closed - Opened by copybara-service[bot] 11 months ago

#338 - Fix typing annotations for get_parameters_overview

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago

#337 - Support sharding for get_parameter_overview

Pull Request - State: open - Opened by copybara-service[bot] 12 months ago

#336 - do not pull in tensorboard's keras callback

Pull Request - State: open - Opened by haohuanw about 1 year ago - 1 comment

#335 - do not pull in tensorboard's keras callback

Pull Request - State: closed - Opened by haohuanw about 1 year ago - 1 comment

#334 - Updates CHANGELOG for v0.0.11

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#333 - Unify the handling of tensor-valued metrics of Std with Average. In particular, this removes the ndim restriction on the Std metric.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#332 - Add unit test passing ShapeDtypeStruct to get_parameter_overview.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#331 - Allow paramter_overview to work with JAX ShapeDtypeStruct.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#330 - Adds Dtype column and total size (in bytes).

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#329 - Make sharding printing work (default fallback to str) for arbitrary sharding types.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago

#328 - Adds "Sharding" column to parameter overview.

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago