Code

added wrt option to nnx.Optimizer#3983

Code

Merged

copybara-service[bot] merged 1 commit intogoogle:maingoogle/flax:mainfrom chiamp:optimizerchiamp/flax:optimizerCopy head branch name to clipboardJun 12, 2024

+111-13

Commits

Commits on Jun 12, 2024

• added wrt option to nnx.Optimizer.update

  chiamp

  committed
  9a40b3f

  Copy full SHA for 9a40b3f