[flax:examples:mnist] Update mnist example to use NNX. #5002
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
copybara-service
bot
force-pushed
the
test_815862166
branch
2 times, most recently
from
00b1d67 to
5d57f27
Compare
copybara-service
bot
changed the title
get_fake_batch and get_apply_fn_and_args methods to train.py.
copybara-service
bot
force-pushed
the
test_815862166
branch
from
5d57f27 to
5c032b4
Compare
samanklesaria
force-pushed
the
test_815862166
branch
2 times, most recently
from
f467b56 to
b263058
Compare
cgarciae
approved these changes
Oct 29, 2025
samanklesaria
force-pushed
the
test_815862166
branch
from
b263058 to
de11539
Compare
Collaborator
|
The current failure is the same as #5054 |
samanklesaria
force-pushed
the
test_815862166
branch
from
de11539 to
f008ef4
Compare
Collaborator
|
I added a check for older versions of python to skip when we can't use tensorflow 2.20. Closes #5054 |
Collaborator
|
@samanklesaria you might have to copy this into a new branch here as this one was created by copybara, these are often temporary and are meant to be oneway. |
PiperOrigin-RevId: 815862166
samanklesaria
force-pushed
the
test_815862166
branch
from
f008ef4 to
9addd9d
Compare
Collaborator
|
Closing in favor of #5064 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
[flax:examples:mnist] Update mnist example to use NNX.