Code

[nnx] add data parallel toy example#4354

Code

Merged

copybara-service[bot] merged 1 commit intomaingoogle/flax:mainfrom nnx-data-parallel-examplegoogle/flax:nnx-data-parallel-exampleCopy head branch name to clipboardNov 8, 2024

+102

Commits

Commits on Nov 8, 2024

• [nnx] add data parallel toy example

  cgarciae

  committed
  a2237ca

  Copy full SHA for a2237ca