Code

feat: add normalize_by_update_norm#958

Code

Merged

copybara-service[bot] merged 5 commits intogoogle-deepmind:maingoogle-deepmind/optax:mainfrom SauravMaheshkar:saurav/scale_by_grad_normCopy head branch name to clipboardJul 9, 2024

+67

Commits

Commits on Jun 25, 2024

• feat: add normalize_by_update_norm

  SauravMaheshkar

  committed
  1528c68

  Copy full SHA for 1528c68

Commits on Jun 28, 2024

• docs: update docstrings

  Show description for 6b5e78f
  

  SauravMaheshkar

  and

  fabianp

  authored
  6b5e78f

  Copy full SHA for 6b5e78f

• feat: add doctests

  SauravMaheshkar

  committed
  c2a7714

  Copy full SHA for c2a7714

• docs: update transformations

  SauravMaheshkar

  committed
  5c01f6f

  Copy full SHA for 5c01f6f

• docs: update transformations API docs

  SauravMaheshkar

  committed
  b849fc6

  Copy full SHA for b849fc6