flax
Add logical axis global context support for NNX
#4350
Merged

Add logical axis global context support for NNX #4350

copybara-service merged 1 commit into google:main from IvyZX:logical-axis
IvyZX
IvyZX IvyZX force pushed 1 year ago
IvyZX Add logical axis global context support for NNX
4d3006af
IvyZX IvyZX force pushed to 4d3006af 1 year ago
cgarciae
cgarciae approved these changes on 2024-11-01
IvyZX IvyZX added pull ready
copybara-service copybara-service merged d31f2906 into main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone