Mesh-TensorFlow: Deep Learning for Supercomputers
Beyond Data and Model Parallelism for Deep Neural Networks