flax
Add tree-mode support for nnx.vmap
#5250
Merged

Add tree-mode support for nnx.vmap #5250

copybara-service merged 1 commit into main from test_868454836
copybara-service
copybara-service copybara-service force pushed from 35529286 to 8ec4cd13 131 days ago
copybara-service copybara-service force pushed from 8ec4cd13 to 0044e493 131 days ago
copybara-service copybara-service force pushed from 0044e493 to de68159b 131 days ago
copybara-service copybara-service force pushed from de68159b to ad198635 131 days ago
Add tree-mode support for nnx.vmap
3d634412
copybara-service copybara-service force pushed from ad198635 to 3d634412 131 days ago
copybara-service copybara-service merged 3d634412 into main 131 days ago
copybara-service copybara-service deleted the test_868454836 branch 131 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone