Code

[nnx] jit constrain object state#3817

Code

Merged

copybara-service[bot] merged 1 commit intomaingoogle/flax:mainfrom nnx-jit-apply-shardinggoogle/flax:nnx-jit-apply-shardingCopy head branch name to clipboardApr 12, 2024

Commits

Commits on Apr 12, 2024

• [nnx] jit constrain object state

  cgarciae

  committed
  8192b96

  Copy full SHA for 8192b96