Update gemma_backbone.py for sharding config.#1491

Merged

mattdangerw merged 3 commits intomasterfrom

qlzh727-patch-2

Mar 14, 2024

Commits on Mar 6, 2024

Update gemma_backbone.py for sharding config.
qlzh727
authored

Commits on Mar 12, 2024

Update unit test and fix format.
qlzh727
committed

Commits on Mar 14, 2024

Update sharding spec for gemma based on gemma training.
qlzh727
committed