Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[nnx] Simplify traversal #4205

Merged
merged 1 commit into from
Sep 26, 2024
Merged

[nnx] Simplify traversal #4205

merged 1 commit into from
Sep 26, 2024

Conversation

cgarciae
Copy link
Collaborator

@cgarciae cgarciae commented Sep 17, 2024

What does this PR do?

Simplifies some of the traversal machinery in preparation for flaxlib.

@cgarciae cgarciae changed the title simplify graph.py Simplify traversal Sep 26, 2024
@cgarciae cgarciae changed the title Simplify traversal [nnx] Simplify traversal Sep 26, 2024
@copybara-service copybara-service bot merged commit 37f643d into main Sep 26, 2024
19 checks passed
@copybara-service copybara-service bot deleted the nnx-remove-registry branch September 26, 2024 18:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants