flax
remove abstracted_axes from nnx.jit
#5132
Merged

remove abstracted_axes from nnx.jit #5132

copybara-service merged 1 commit into main from test_842278010
copybara-service
copybara-service copybara-service force pushed from 33a164f3 to 2abe568a 155 days ago
remove abstracted_axes from nnx.jit
b3e9b449
copybara-service copybara-service force pushed from 2abe568a to b3e9b449 155 days ago
copybara-service copybara-service merged b3e9b449 into main 155 days ago
copybara-service copybara-service deleted the test_842278010 branch 155 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone