flax
Add option to load checkpoints with transposed Gating Einsum.
#4597

Merged

Add option to load checkpoints with transposed Gating Einsum. #4597

copybara-service merged 1 commit into main from test_733277427

copybara-service force pushed from ca99b449 to bf6a4118 1 year ago

copybara-service force pushed from bf6a4118 to 189f03dc 1 year ago

copybara-service force pushed from 189f03dc to b4d39a62 1 year ago

copybara-service force pushed from b4d39a62 to 81277f8e 1 year ago

Add option to load checkpoints with transposed Gating Einsum.

ceb7aedd

copybara-service force pushed from 81277f8e to ceb7aedd 1 year ago

copybara-service merged ceb7aedd into main 1 year ago

copybara-service deleted the test_733277427 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone