Give AlbumentationsX a star on GitHub — it powers this leaderboard
An implementation of training for GPT2, supports TPUs