flax
Update mnist example to use NNX (clone of #5064)
#5188

Merged

Update mnist example to use NNX (clone of #5064) #5188

copybara-service merged 1 commit into google:main from samanklesaria:pr/5002_copy

Update mnist example to use NNX.

0ef9abb2

samanklesaria requested a review from

cgarciae 117 days ago

IvyZX approved these changes on 2026-01-13

IvyZX added pull ready

IvyZX requested changes on 2026-01-13

copybara-service merged 3da2276b into main 116 days ago

Reviewers

IvyZX

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone