mesh-tensorflow
tensorflow-gpu>=1.15
