Comments (3)
I re-implemented Shampoo optimizer based on your linked code and deployed the package (v2.3.0)! If there're any ideas and requests, feel free to leave a comment or open another issue :)
thanks!
from pytorch_optimizer.
I will give it a try.
from pytorch_optimizer.
thanks for the reference and suggestion!
I agree that the JAX implementation is faster than Pytorch, however, the project's goal is specifically to implement the optimizer using Pytorch. Despite this, there are features that have yet to be developed, such as Grafting and Partitioning. I'll add these features to improve the Shampoo optimizer.
best regard :)
from pytorch_optimizer.
Related Issues (20)
- Prodigy: An Expeditiously Adaptive Parameter-Free Learner HOT 2
- LOMO: LOw-Memory Optimization HOT 1
- Can a variant of Lion named Tiger be added to your package? HOT 2
- sophiah bug HOT 5
- sophiah in https://github.com/booydar/LM-RMT HOT 8
- Adding the CAME optimizer HOT 2
- Lookahead is not a subclass of torch.optim.Optimizer HOT 4
- Empty Docs Sections HOT 6
- Request to add 4-bit AdamW HOT 3
- ipex failed for Adan from pytorch_optimizer HOT 1
- Improvement to SAM: SAM as an Optimal Relaxation of Bayes HOT 1
- FR: Sharpness-Aware Minimization Revisited: Weighted Sharpness as a Regularization Term (WSAM) HOT 2
- Ranger21 has undocumented required arguments HOT 3
- [Feature request]REX LR scheduler HOT 2
- Aida optimizer HOT 3
- GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection HOT 1
- Adalite HOT 1
- ScheduleFree
- Entropy-MCMC: Sampling from flat basins with ease
- Ranger sign inversion HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from pytorch_optimizer.