flax
[nnx] TrainState uses struct
#3788
Merged

[nnx] TrainState uses struct #3788

copybara-service merged 1 commit into main from nnx-train-state-struct
cgarciae
cgarciae cgarciae force pushed 2 years ago
cgarciae cgarciae force pushed 2 years ago
review-notebook-app
codecov-commenter
cgarciae cgarciae force pushed 2 years ago
cgarciae cgarciae force pushed 2 years ago
chiamp
chiamp approved these changes on 2024-03-28
cgarciae cgarciae force pushed 2 years ago
cgarciae cgarciae force pushed 2 years ago
cgarciae cgarciae added pull ready
cgarciae [nnx] TrainState uses struct
51b97edd
cgarciae cgarciae force pushed to 51b97edd 2 years ago
copybara-service copybara-service merged 9c4c8f58 into main 2 years ago
copybara-service copybara-service deleted the nnx-train-state-struct branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone