flax
Add tree-mode support for nnx.custom_vjp
#5264
Merged

Add tree-mode support for nnx.custom_vjp #5264

copybara-service merged 1 commit into main from test_870359621
copybara-service
copybara-service copybara-service force pushed from ec9f8a16 to b80e9352 23 days ago
Add tree-mode support for nnx.custom_vjp
c9252176
copybara-service copybara-service force pushed from b80e9352 to c9252176 23 days ago
copybara-service copybara-service merged c9252176 into main 23 days ago
copybara-service copybara-service deleted the test_870359621 branch 23 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone