Star AlbumentationsX on GitHub — it powers this leaderboard
PKUFlyingPig/alpa
Auto parallelization for large-scale neural networks