Rose

无状态优化器,省内存,稳收敛,直接替换 Adam

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 5 👀 关注 4
GitHub ↗
GitHub 星标 ★ 79
维护状态 较活跃
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0许可证,可自由使用,无付费版本。

项目预览

Rose 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队MatthewK78
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0许可证,可自由使用,无付费版本。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型adam-optimizer,ai,artificial-intelligence,deep-learning,diffusion-models,fine-tuning,gpt,gradient-descent,large-language-models,llm,lora,machine-learning,optimizer,optimizer-algorithms,optimizers,pretraining,pytorch,sgd,transformers
GitHub 星标★ 79
上线时间2026-04-17 00:00:00
代码更新2026-10-09
仓库体积95 KB
本站数据更新2026-11-08
维护状态较活跃
收录时间2026-08-09
浏览次数8

项目介绍

Rose 是一个基于 PyTorch 的优化器,提出了一种名为 Range-Of-Slice Equilibration 的优化方法。它通过范围归一化的梯度更新来实现无状态优化,旨在解决传统优化器(如 Adam)依赖动量状态、内存占用高且对超参数敏感的问题。Rose 的核心思想是将梯度按切片范围进行均衡化,使每个参数的更新步长与其梯度分布相匹配,从而在无需维护额外状态的情况下保持稳定的收敛性能。该项目的目标是提供一种轻量级、内存高效的优化器替代方案,特别适用于大规模模型训练或资源受限的环境。当前项目处于早期阶段,代码简洁,易于集成到现有 PyTorch 训练流程中。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 79

核心亮点

  • 无状态设计,显著降低优化器内存占用,适合大模型训练
  • 梯度范围归一化,减少对学习率等超参数的敏感度
  • 代码简洁,基于 PyTorch,易于集成和实验

不足之处

  • 项目处于早期阶段,文档和社区支持尚不完善
  • 缺乏与主流优化器(如 AdamW)在多种任务上的全面基准对比

适用场景

  • 大规模模型训练时降低内存开销
  • 超参数调优成本高的场景,希望减少学习率调节
  • 边缘设备或资源受限环境下的模型微调

替代项目

AdamW、Lion、Adafactor

上一篇:VisionFoundry

下一篇:DuQuant

同类项目推荐

verl-omni 开源

给扩散和全模态模型做 RL 后训练,一套框架跑通

Multimodal RL training framework for diffusion & omni models

★ 1471 2026-10-12
Awesome-MLLM-Tuning 开源

多模态大模型微调不丢本事的论文与思路合集

Multimodal Large Language Model (MLLM) Tuning Survey: Keeping Yourself is Important ···

★ 102 2026-10-28
nanotron 开源

极简代码跑通千亿模型3D并行训练

Minimalistic large language model 3D-parallelism training

★ 2854 2026-09-25
MirrorFlow 开源

对话喂出来的数字分身,越聊越像你自己。

从对话数据到训练:数字分身 + 模型蒸馏 From Dialogue Data to Training Closed-Loop: D···

★ 73 2026-08-09
AI-Toolkit-Easy-Install 开源

Windows 上一键装好 AI 微调工具,免折腾环境直接开训

One-click Portable Windows installation of 'AI-Toolkit by Ostris'

★ 541 2026-08-09
lightning-hydra-template 开源

开箱即用的深度学习实验框架,配置驱动,快速复现和扩展。

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡···

★ 5380 2026-08-09
Medusa 开源

给大模型加几个解码头,生成速度翻倍,不用改架构

Medusa: Simple Framework for Accelerating LLM Generation with Multiple Decoding Head···

★ 2774 2026-08-09
CausalPFN 开源

用上下文学习,秒级估算因果效应,告别复杂建模

CausalPFN: Amortized Causal Effect Estimation via In-Context Learning

★ 137 2026-08-09