flax
Remove nnx.split_rngs call wrapping nnx.scan in linen to nnx tutorial
#5160

Merged

Remove nnx.split_rngs call wrapping nnx.scan in linen to nnx tutorial #5160

copybara-service merged 1 commit into google:main from samanklesaria:issues/5158

Remove nnx.split_rngs call wrapping nnx.scan

1aca395b

cgarciae marked this pull request as draft 184 days ago

cgarciae marked this pull request as ready for review 184 days ago

cgarciae approved these changes on 2026-01-05

cgarciae added pull ready

copybara-service merged f3506885 into main 184 days ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone