flax
Call user-defined variable transforms before determining axis size in nn.vmap.
#4026

Merged

Call user-defined variable transforms before determining axis size in nn.vmap. #4026

copybara-service merged 1 commit into main from test_646334059

copybara-service force pushed 2 years ago

Call user-defined variable transforms before determining axis size in…

e5cb2f76

copybara-service force pushed to e5cb2f76 2 years ago

copybara-service merged e5cb2f76 into main 2 years ago

copybara-service deleted the test_646334059 branch 2 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone