Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 5, 2025

• Update clu to use inspect.get_annotations(cls) over cls.__annotations__

  Show description for a8152eb
  

  CLU Authors

  authored and

  copybara-github

  committed
  a8152eb

  Copy full SHA for a8152eb

Commits on Jan 30, 2025

• Fix when variable is None for include_stats is True or False

  Show description for 43acbbd
  

  CLU Authors

  authored and

  copybara-github

  committed
  43acbbd

  Copy full SHA for 43acbbd

Commits on Jan 28, 2025

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

  Show description for 77b0602
  

  CLU Authors

  authored and

  copybara-github

  committed
  77b0602

  Copy full SHA for 77b0602

Commits on Jan 14, 2025

• No public description

  Show description for b4e47eb
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  b4e47eb

  Copy full SHA for b4e47eb

• Fix handling of empty parameters in parameter_overview.

  Show description for c139e6a
  

  andresusanopinto

  authored and

  copybara-github

  committed
  c139e6a

  Copy full SHA for c139e6a

Commits on Aug 21, 2024

• Add point clouds summary writer to kauldron interface with tensorboard and jaxboard.

  Show description for 307b0bc
  

  CLU Authors

  authored and

  copybara-github

  committed
  307b0bc

  Copy full SHA for 307b0bc

Commits on Aug 12, 2024

• Allow running profiler on specific steps.

  Show description for 7c22ddf
  

  CLU Authors

  authored and

  copybara-github

  committed
  7c22ddf

  Copy full SHA for 7c22ddf

Commits on Jul 19, 2024

• Rollback add point clouds summary writer to tensorboard interface and metric writer.

  Show description for b64aa29
  

  CLU Authors

  authored and

  copybara-github

  committed
  b64aa29

  Copy full SHA for b64aa29

Commits on Jul 17, 2024

• Add point clouds summary writer to tensorboard interface and metric writer.

  Show description for d090965
  

  CLU Authors

  authored and

  copybara-github

  committed
  d090965

  Copy full SHA for d090965

Commits on May 29, 2024

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

  Show description for 92334e9
  

  mbrukman

  authored and

  copybara-github

  committed
  92334e9

  Copy full SHA for 92334e9

Commits on Apr 24, 2024

• Fix bug in preprocess_spec.py: Remove list() that fails if tensor rank is unknown.

  Show description for b172f5a
  

  mjlm

  authored and

  copybara-github

  committed
  b172f5a

  Copy full SHA for b172f5a

Commits on Apr 17, 2024

• No public description

  Show description for 613dcbb
  

  Conchylicultor

  authored and

  copybara-github

  committed
  613dcbb

  Copy full SHA for 613dcbb

Commits on Apr 9, 2024

• Release v0.0.12.

  Show description for d381738
  

  Marvin182

  authored and

  copybara-github

  committed
  d381738

  Copy full SHA for d381738

Commits on Apr 5, 2024

• Change deprecated jax.tree_map to avoid warnings:

  Show description for 5e7a1e9
  

  CLU Authors

  authored and

  copybara-github

  committed
  5e7a1e9

  Copy full SHA for 5e7a1e9

• 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 t…

  Show description for 873ad23
  

  chiamp

  authored and

  copybara-github

  committed
  873ad23

  Copy full SHA for 873ad23

Commits on Mar 14, 2024

• No public description

  Show description for c50acb7
  

  jpuigcerver

  authored and

  copybara-github

  committed
  c50acb7

  Copy full SHA for c50acb7

Commits on Mar 4, 2024

• Fix typing annotations for get_parameters_overview

  Show description for 0b961e5
  

  Conchylicultor

  authored and

  copybara-github

  committed
  0b961e5

  Copy full SHA for 0b961e5

Commits on Feb 19, 2024

• Updates CHANGELOG for v0.0.11

  Show description for eed40a1
  

  andsteing

  authored and

  copybara-github

  committed
  eed40a1

  Copy full SHA for eed40a1

Commits on Feb 1, 2024

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

  Show description for 1368e52
  

  lzlarryli

  authored and

  copybara-github

  committed
  1368e52

  Copy full SHA for 1368e52

Commits on Jan 3, 2024

• Add unit test passing ShapeDtypeStruct to get_parameter_overview.

  Show description for f30bc44
  

  jpuigcerver

  authored and

  copybara-github

  committed
  f30bc44

  Copy full SHA for f30bc44

• Allow paramter_overview to work with JAX ShapeDtypeStruct.

  Show description for 574d4c9
  

  jpuigcerver

  authored and

  copybara-github

  committed
  574d4c9

  Copy full SHA for 574d4c9

Commits on Dec 22, 2023

• Adds Dtype column and total size (in bytes).

  Show description for 8a01f21
  

  andsteing

  authored and

  copybara-github

  committed
  8a01f21

  Copy full SHA for 8a01f21

• Make sharding printing work (default fallback to __str__) for arbitrary sharding types.

  Show description for d61576a
  

  Lucas Beyer

  authored and

  copybara-github

  committed
  d61576a

  Copy full SHA for d61576a

Commits on Dec 21, 2023

• No public description

  Show description for 27d59a4
  

  andsteing

  authored and

  copybara-github

  committed
  27d59a4

  Copy full SHA for 27d59a4

Commits on Dec 19, 2023

• No public description

  Show description for 1aa3773
  

  CLU Authors

  authored and

  copybara-github

  committed
  1aa3773

  Copy full SHA for 1aa3773

Commits on Nov 20, 2023

• Explicitly handle None in model parameters

  Show description for 7ba2a9d
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  7ba2a9d

  Copy full SHA for 7ba2a9d

Commits on Oct 4, 2023

• Remove numpy version pin

  Show description for be91a31
  

  CLU Authors

  authored and

  copybara-github

  committed
  be91a31

  Copy full SHA for be91a31

Commits on Oct 2, 2023

• Updates version to v0.0.10

  Show description for 69845bd
  

  andsteing

  authored and

  copybara-github

  committed
  69845bd

  Copy full SHA for 69845bd

Commits on Sep 22, 2023

• binds step argument for ProfileAllHosts

  Show description for ba71bff
  

  Conchylicultor

  authored and

  copybara-github

  committed
  ba71bff

  Copy full SHA for ba71bff

Commits on Sep 7, 2023

• Use the mean()/std() method from the value object instead np.mean or np.std. This enables logging even for parameter nodes that aren't numpy arrays but instances of custom classes.

  Show description for 38caf2b
  

  MostafaDehghani

  authored and

  copybara-github

  committed
  38caf2b

  Copy full SHA for 38caf2b

Commits on Aug 25, 2023

• Remove TensorFlow import.

  Show description for 17c4624
  

  Marvin182

  authored and

  copybara-github

  committed
  17c4624

  Copy full SHA for 17c4624

Commits on Aug 24, 2023

• param_overview: Make device -> host transfer explicit instead of implicit.

  Show description for ee7eae2
  

  Lucas Beyer

  authored and

  copybara-github

  committed
  ee7eae2

  Copy full SHA for ee7eae2

• Make parameter_overview work on global arrays.

  Show description for 1ae207c
  

  Lucas Beyer

  authored and

  copybara-github

  committed
  1ae207c

  Copy full SHA for 1ae207c

Commits on Aug 23, 2023

• No public description

  Show description for efa68db
  

  hawkinsp

  authored and

  copybara-github

  committed
  efa68db

  Copy full SHA for efa68db

Commits on Aug 14, 2023

• No public description

  Show description for c92475e
  

  hawkinsp

  authored and

  copybara-github

  committed
  c92475e

  Copy full SHA for c92475e

Pagination

Previous

Next