flax
[nnx] revive TrainState toy example
#4226
Merged

[nnx] revive TrainState toy example #4226

cgarciae
cgarciae [nnx] revive TrainState toy example
2d6aa629
superbobry
superbobry approved these changes on 2024-10-08
cgarciae cgarciae added pull ready
copybara-service copybara-service merged 62a3414f into main 1 year ago
copybara-service copybara-service deleted the nnx-train-state-toy-example branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone