跳到正文
arXiv Artificial Intelligence· Kathan Shah·· 19 小时前AI 评分26

Ember:面向 Token 接口的千字节状态优化器

Ember: A Kilobyte-State Optimizer for Token Interfaces

AI 导读

Ember 是一款针对嵌入表和 LM-head 矩阵的轻量优化器,仅需 O(V+D) 优化器显存而非 Adam 的 O(2VD),且无需分片 token 表优化器状态。

来源:arXiv Artificial Intelligence · arxiv.org