PR #3782 Add `nnx.training`

Add `nnx.training` #3782

copybara-service merged 1 commit into google:main from chiamp:nnx_module_subclass

chiamp assigned

chiamp 2 years ago

chiamp marked this pull request as draft 2 years ago

chiamp force pushed 2 years ago

chiamp force pushed 2 years ago

chiamp force pushed to 297c3889 2 years ago

chiamp requested a review from

cgarciae 2 years ago

cgarciae commented on 2024-03-22

chiamp force pushed from 297c3889 2 years ago

chiamp force pushed 2 years ago

cgarciae commented on 2024-03-26

chiamp force pushed 2 years ago

cgarciae force pushed 2 years ago

chiamp force pushed 2 years ago

cgarciae force pushed to a294f849 2 years ago

chiamp force pushed from a294f849 2 years ago

chiamp force pushed 2 years ago

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

cgarciae approved these changes on 2024-04-01

cgarciae added pull ready

chiamp force pushed 2 years ago

have TrainState subclass nnx.Module

543686b1

chiamp force pushed to 543686b1 2 years ago

copybara-service merged 5d9fd9fb into main 2 years ago

chiamp deleted the nnx_module_subclass branch 2 years ago

Reviewers

cgarciae

Assignees

chiamp

Labels

pull ready

Milestone

No milestone