(WIP) Distributed Training for Large Scale Transformer (5/6) - Expert Parallelism (EP)