Star AlbumentationsX on GitHub — it powers this leaderboard
tensorflow/mesh
Mesh TensorFlow: Model Parallelism Made Easier