Repository navigation
pytorch-optimizer v3.11.0
What's Changed
- [Docs] Prepend changelog for v3.10.1 by @kozistr in #509
- [Feature] Implement
Magmaoptimizer by @kozistr in #514 - [Fix] Keep
SignSGDmomentum buffer across steps by @11NOel11 in #516 - [Fix] Apply weight decay in
SignSGDby @11NOel11 in #518 - [Fix] Match
foreachand per-parameter rounding inSignSGDandTigerby @11NOel11 in #520 - Fix AdaSmooth NaN on zero-initialized parameters by @winklemad in #522
- Fix AdaShift NaN on the first update with keep_num > 1 by @winklemad in #523
- Fix A2Grad diverging to NaN from an inverted running-mean weight by @winklemad in #524
- Fix M-SVAG NaN from debiasing moments by betastep instead of 1 - betastep by @winklemad in #525
- [Fix]
rhocalculation forM-SVAGoptimizer by @kozistr in #529 - docs: migrate to Zensical and automate changelogs by @kozistr in #532
- Raise Shampoo factors to -1/(2k) by @shaneraphel in #530
- Do not start an AdamMini collective without a process group by @shaneraphel in #534
- Keep MARS last_grad stable across the next backward by @shaneraphel in #533
- Divide the DAdaptAdam distance estimate by sk_l1 by @shaneraphel in #538
- Store the first PID gradient for the next derivative by @shaneraphel in #539
- [Fix] scale StableAdamW after the second moment is updated by @shaneraphel in #544
- [Fix] advance the Prodigy step counter once per update by @shaneraphel in #546
- [Fix] keep LaProp's centered second moment out of the stored average by @shaneraphel in #547
- [Fix] bias-correct only the fast AdEMAMix moment by @shaneraphel in #549
- [Fix] lift Alice subspace vectors back to the parameter shape by @shaneraphel in #550
- Keep ApolloDQN stable weight decay inside one parameter by @shaneraphel in #535
- [Fix] lower DAdaptAdam test learning rate by @kozistr in #552
- Fix MADGRAD adding eps to the learning rate by @winklemad in #526
- Apply the documented
maximizeoption in optimizers that ignored it by @winklemad in #527 - Fix Adai ignoring
maximize(gradient negated twice) by @winklemad in #528 - [Fix] Match FAdam adaptive epsilon to the paper by @kozistr in #557
- [Fix] Correct Schedule-Free wrapper state and AdamMini GQA updates by @kozistr in #558
- [Fix] Apply MARS 1D weight decay by @kozistr in #559
- [Fix] Respect FlashAdamW learning-rate decay mode by @kozistr in #561
- [Fix] Handle mixed dtypes in StableAdamW foreach updates by @kozistr in #560
- [Fix] Honor maximize in StableAdamW scalar updates by @kozistr in #562
- [Fix] Honor maximize in LOMO updates by @kozistr in #563
- [Fix] Initialize LARS momentum from zero by @kozistr in #564
- [Fix] Include decay in the LARS trust ratio by @kozistr in #565
- [Fix] Initialize GrokFast EMA for maximize by @kozistr in #566
- [Fix] Use beta0 in Ranger21 positive-negative momentum by @kozistr in #568
- [Fix] Reject invalid schedule-free momentum values by @kozistr in #567
- [Fix] Preserve Ranger21 second-moment maximum by @kozistr in #569
- Move docs requirements file under docs/ by @kozistr in #570
- Honor Ranger21 gradient processing options by @kozistr in #571
- Clarify pull request title guidance by @kozistr in #572
- Apply coupled LAMB decay within the trust ratio by @kozistr in #573
- Match NovoGrad's first update and step size to the paper by @kozistr in #574
- [Fix] Correct Hessian sampling and optimizer edge cases by @kozistr in #575
- [Fix] Preserve base optimizer state in SAM checkpoints by @kozistr in #576
- [Fix] Remap TRAC references when loading checkpoints by @kozistr in #579
- [Fix] Project OrthoGrad gradients after closure evaluation by @kozistr in #583
- [Fix] Keep Lookahead slow momentum in separate storage by @kozistr in #582
- [Fix] Honor zero weight decay in create_optimizer by @kozistr in #580
- [Fix] Apply StableAdamW coupled decay before moment updates by @kozistr in #581
- [Fix] Apply BCE focal class weights and elementwise reduction by @AHMETHAKANBEZIR1 in #585
- [Fix] Stabilize OrthoGrad and FTRL GPU training by @kozistr in #586
- [Build] Add manual release workflow by @kozistr in #587
- Accumulate the AdaNorm norm average and guard zero gradients by @shaneraphel in #536
- [Fix] take the Ano second-moment sign before the decay by @shaneraphel in #553
New Contributors
- @11NOel11 made their first contribution in #516
- @winklemad made their first contribution in #522
- @shaneraphel made their first contribution in #530
- @AHMETHAKANBEZIR1 made their first contribution in #585
Full Changelog: v3.10.1...v3.11.0