flax
[nnx] jit constrain object state
#3817
Merged

[nnx] jit constrain object state #3817

copybara-service merged 1 commit into main from nnx-jit-apply-sharding
cgarciae
cgarciae cgarciae force pushed 2 years ago
codecov-commenter
cgarciae cgarciae force pushed to a88fbbba 2 years ago
cgarciae cgarciae marked this pull request as ready for review 2 years ago
superbobry
superbobry commented on 2024-04-12
superbobry
superbobry commented on 2024-04-12
superbobry
superbobry approved these changes on 2024-04-12
cgarciae cgarciae force pushed from a88fbbba 2 years ago
cgarciae cgarciae added pull ready
cgarciae [nnx] jit constrain object state
8192b966
cgarciae cgarciae force pushed to 8192b966 2 years ago
copybara-service copybara-service merged 2438518b into main 2 years ago
copybara-service copybara-service deleted the nnx-jit-apply-sharding branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone