flax
Add tree-mode support to nnx.{cond,switch,eval_shape,checkify}
#5252

Merged

Add tree-mode support to nnx.{cond,switch,eval_shape,checkify} #5252

copybara-service merged 1 commit into main from test_870184940

copybara-service force pushed from ba6b0976 to 5f2d90d2 1 day ago

copybara-service force pushed from 5f2d90d2 to 469d0534 1 day ago

copybara-service force pushed from 469d0534 to 1203e8a4 1 day ago

copybara-service force pushed from 1203e8a4 to f9ea562f 1 day ago

copybara-service force pushed from f9ea562f to e35d833f 1 day ago

Add tree-mode support to nnx.{cond,switch,eval_shape,checkify}

761a27d3

copybara-service force pushed from e35d833f to 761a27d3 1 day ago

copybara-service merged 761a27d3 into main 1 day ago

copybara-service deleted the test_870184940 branch 1 day ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone