Skip to content

stochastic rounding of momentum updates#20

Draft
dxqb wants to merge 3 commits into
Koratahiu:mainfrom
dxqb:SR_momentum_upstream
Draft

stochastic rounding of momentum updates#20
dxqb wants to merge 3 commits into
Koratahiu:mainfrom
dxqb:SR_momentum_upstream

Conversation

@dxqb

@dxqb dxqb commented Mar 17, 2026

Copy link
Copy Markdown
Contributor

only for default AdamW
experimental code, only for testing purposes

print debug message to make sure it's enabled, until there is a UI to config it
@Koratahiu

Copy link
Copy Markdown
Owner

This was added in #19 as a new state system that accepts bf16_sr.
Thanks for testing and proving its usefulness; I will mention you in the changelog!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants