PR #3933 [nnx] improve update context

[nnx] improve update context #3933

copybara-service merged 2 commits into main from nnx-improve-update-context

cgarciae force pushed 2 years ago

cgarciae marked this pull request as ready for review 2 years ago

chiamp commented on 2024-05-23

chiamp approved these changes on 2024-05-23

cgarciae force pushed 2 years ago

[nnx] improve update context

eeab2142

cgarciae force pushed to eeab2142 2 years ago

Update flax/experimental/nnx/nnx/graph.py

57dd8489

cgarciae added pull ready

copybara-service merged 9b90e98b into main 2 years ago

copybara-service deleted the nnx-improve-update-context branch 2 years ago

Reviewers

chiamp

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone