flax
Add tree-mode-only nnx.{vjp,jvp,jit_partial}
#5268

Merged

Add tree-mode-only nnx.{vjp,jvp,jit_partial} #5268

copybara-service merged 1 commit into main from test_873087034

copybara-service force pushed from d20e3102 to a789e7ba 15 days ago

copybara-service force pushed from a789e7ba to a217064c 15 days ago

copybara-service force pushed from a217064c to 4de5d999 15 days ago

copybara-service force pushed from 4de5d999 to 9405e3e6 15 days ago

copybara-service force pushed from 9405e3e6 to 4a2121c3 15 days ago

copybara-service force pushed from 4a2121c3 to 0b1b5375 15 days ago

copybara-service force pushed from 0b1b5375 to 1c031251 14 days ago

copybara-service force pushed from 1c031251 to 354f4b76 13 days ago

copybara-service force pushed from 354f4b76 to f76e1627 13 days ago

Add tree-mode-only nnx.{vjp,jvp,jit_partial}

9a1bc4eb

copybara-service force pushed from f76e1627 to 9a1bc4eb 13 days ago

copybara-service merged 9a1bc4eb into main 13 days ago

copybara-service deleted the test_873087034 branch 13 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone