flax
revise axes_scan to flatten argument pytrees only once
#4772

Merged

revise axes_scan to flatten argument pytrees only once #4772

copybara-service merged 1 commit into main from test_768154467

copybara-service force pushed from 3184e8e7 to 77b0c222 220 days ago

revise axes_scan to flatten argument pytrees only once

893a6602

copybara-service force pushed from 77b0c222 to 893a6602 220 days ago

copybara-service merged 893a6602 into main 220 days ago

copybara-service deleted the test_768154467 branch 220 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone