Add option to load checkpoints with transposed Gating Einsum. · google/flax@ceb7aed

Canceling since a higher priority waiting request for 'Flax - Test-refs/heads/main' exists

The operation was canceled.

The ubuntu-20.04 runner image will be end of life on 2025-04-01. Jobs using the ubuntu-20.04 image should be updated to ubuntu-22.04 or ubuntu-24.04. For more details, see https://github.com/actions/runner-images/issues/11101