Adam-mini. Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793