v0.2.2
[embedding] rename FreqAwareEmbedding -> CachedEmbedding (#1699)
[zero] add zero wrappers (#2523)
[hotfix] fix zero ddp warmup check (#2545)
[hotfix] fix lightning error (#2529)
[ddp] ColoDDP uses bucket all-reduce (#1177)
[gemini] update the gpt example (#2527)