[Updated on 2022-03-13: add expert choice routing.] [Updated on 2022-06-10]: Greg and I wrote a shorted and upgraded version of this post, published on OpenAI Blog: “Techniques for Training Large Neural Networks”
Read the original at Lil'Log (Lilian Weng): How to Train Really Large Models on Many GPUs?
Source: https://lilianweng.github.io/posts/2021-09-25-train-large/