Davide Cifarelli's picture

Davide Cifarelli

davide221

AI & ML interests

codeLLM, reasoning & interpretability

Recent Activity

Organizations

Safurai's profile picture Blog-explorers's profile picture positivethoughts's profile picture Klara HDL's profile picture Elementar srl's profile picture Klara Labs's profile picture

davide221's activity

upvoted an article 6 months ago
reacted to yushun0410's post with πŸ”₯ 7 months ago
view post
Post
4624
Hi Huggingfacers!

Thrilled to introduce Adam-mini, an optimizer that achieves on-par or better performance than AdamW with 45% to 50% less memory footprint. Adam-mini can also achieve 49.5% higher throughput than AdamW on Llama2-7B pre-training.

The design of Adam-mini is inspired by certain Hessian structures we observed on Transformers.

Feel free to try it out! Try switching to Adam-mini with the same hyperparams of AdamW, it would work with only half memory. Hope Adam-mini can help save time, cost, and energy in your tasks!

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

Code: https://github.com/zyushun/Adam-mini

  • 1 reply
Β·