Mesh TensorFlow - Model Parallelism Made Easier
