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 1 day ago

cgarciae marked this pull request as ready for review 1 day ago

cgarciae approved these changes on 2026-01-05

cgarciae added pull ready

copybara-service merged f3506885 into main 1 day ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone