GitHub repository: tensorflow/mesh
Primary language: Python
Description: Mesh TensorFlow: Model Parallelism Made Easier