flax
Add tree-mode support to nnx.{while,fori}_loop
#5263

Merged

Add tree-mode support to nnx.{while,fori}_loop #5263

copybara-service merged 1 commit into main from test_870359623

copybara-service force pushed from 28289219 to 0a73e714 14 days ago

copybara-service force pushed from 0a73e714 to 732cb545 14 days ago

copybara-service force pushed from 732cb545 to 156e8940 14 days ago

Add tree-mode support to nnx.{while,fori}_loop

e108ab89

copybara-service force pushed from 156e8940 to e108ab89 14 days ago

copybara-service merged e108ab89 into main 14 days ago

copybara-service deleted the test_870359623 branch 14 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone