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 1 year ago

cgarciae marked this pull request as ready for review 1 year ago

chiamp commented on 2024-05-23

chiamp approved these changes on 2024-05-23

cgarciae force pushed 1 year ago

[nnx] improve update context

eeab2142

cgarciae force pushed to eeab2142 1 year ago

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

57dd8489

cgarciae added pull ready

copybara-service merged 9b90e98b into main 1 year ago

copybara-service deleted the nnx-improve-update-context branch 1 year ago

Reviewers

chiamp

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone