flax
Add tree-mode support for nnx.remat
#5242
Merged

Add tree-mode support for nnx.remat #5242

copybara-service merged 1 commit into main from test_868394505
copybara-service
copybara-service copybara-service force pushed from c60aa966 to 4b200dcf 117 days ago
copybara-service copybara-service force pushed from 4b200dcf to cd9cc875 116 days ago
Add tree-mode support for nnx.remat
aed27f67
copybara-service copybara-service force pushed from cd9cc875 to aed27f67 116 days ago
copybara-service copybara-service merged aed27f67 into main 116 days ago
copybara-service copybara-service deleted the test_868394505 branch 116 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone