flax
ceb7aedd - Add option to load checkpoints with transposed Gating Einsum.

Commit

1 year ago

Add option to load checkpoints with transposed Gating Einsum. PiperOrigin-RevId: 734225807

References

Author

casaro

Committer

a-googler

Parents