Give AlbumentationsX a star on GitHub — it powers this leaderboard
Dynamic tensor rematerialization, implemented as a pytorch fork