transformers

f1c81d6b - [Flax] ViT training example (#12300)

[Flax] ViT training example (#12300) * begin script * clean example, add readme * update readme * remove decay mask * remove masking * update readme & make flake happy