GitHub repository: DifferentiableUniverseInitiative/mesh
Description: Mesh TensorFlow: Model Parallelism Made Easier