Give AlbumentationsX a star on GitHub — it powers this leaderboard
Adds GaLore style projection wrappers to optax optimizers