GitHub repository: min-xu-ai/mesh
Description: Mesh TensorFlow: Model Parallelism Made Easier