Code

[nnx] revive TrainState toy example#4226

Code

Merged

copybara-service[bot] merged 1 commit intomaingoogle/flax:mainfrom nnx-train-state-toy-examplegoogle/flax:nnx-train-state-toy-exampleCopy head branch name to clipboardOct 8, 2024

+123

Commits

Commits on Sep 24, 2024

• [nnx] revive TrainState toy example

  cgarciae

  committed
  2d6aa62

  Copy full SHA for 2d6aa62