PR #4811 [nnx] add flax_pytree_module flag

[nnx] add flax_pytree_module flag #4811

copybara-service merged 1 commit into main from nnx-pytree-modules-2

cgarciae force pushed from 947101ed to a1299624 278 days ago

cgarciae force pushed from a1299624 to e384afd0 278 days ago

cgarciae force pushed from e384afd0 to 63ff15dd 278 days ago

cgarciae marked this pull request as ready for review 276 days ago

cgarciae force pushed from 63ff15dd to da0e9039 275 days ago

cgarciae force pushed from da0e9039 to 7d11a0ff 266 days ago

cgarciae force pushed from 7d11a0ff to bcd70719 266 days ago

cgarciae force pushed from bcd70719 to f0a69a38 266 days ago

cgarciae added pull ready

cgarciae force pushed from f0a69a38 to c2e18bcf 266 days ago

[nnx] add flax_pytree_module flag

50e043c2

cgarciae force pushed from c2e18bcf to 50e043c2 266 days ago

copybara-service merged ae208ee7 into main 264 days ago

copybara-service deleted the nnx-pytree-modules-2 branch 264 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone