flax icon indicating copy to clipboard operation
flax copied to clipboard
verified publisher icon google

Update mnist example to use NNX (clone of #5002)

Open samanklesaria opened this issue 5 months ago • 1 comments

Update mnist example to use NNX. Closes #5054. Same as #5002, only not coming from copybara (so potentially easier to merge with changers).

samanklesaria avatar Oct 31 '25 17:10 samanklesaria

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

review-notebook-app[bot] avatar Oct 31 '25 17:10 review-notebook-app[bot]