optimizer & lr scheduler & loss function collections in PyTorch
☆391Mar 1, 2026Updated this week
Alternatives and similar repositories for pytorch_optimizer
Users that are interested in pytorch_optimizer are comparing it to the libraries listed below
Sorting:
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Prodigy and Schedule-Free, together at last.☆87Sep 27, 2025Updated 5 months ago
- stochastic bfloat16 based optimizer library☆21Dec 4, 2024Updated last year
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆983Jan 30, 2024Updated 2 years ago
- Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.☆12Jun 5, 2024Updated last year
- [ACL 2023] The official implementation of "CAME: Confidence-guided Adaptive Memory Optimization"☆96Mar 22, 2025Updated 11 months ago
- D-Adaptation for SGD, Adam and AdaGrad☆529Jan 22, 2025Updated last year
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆808Jun 8, 2025Updated 8 months ago
- Ranger deep learning optimizer rewrite to use newest components☆341Feb 18, 2024Updated 2 years ago
- A collection of niche / personally useful PyTorch optimizers with modified code.☆27Oct 25, 2025Updated 4 months ago
- Permutation invariant training in PyTorch☆13Oct 2, 2020Updated 5 years ago
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.☆624Feb 2, 2026Updated last month
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆453May 13, 2025Updated 9 months ago
- ☆21Jan 23, 2024Updated 2 years ago
- Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training☆36Jun 20, 2025Updated 8 months ago
- Multidimensional indexing for tensors☆137Jul 17, 2023Updated 2 years ago
- Muon is an optimizer for hidden layers in neural networks☆2,350Jan 19, 2026Updated last month
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,963Feb 21, 2024Updated 2 years ago
- Collect optimizer related papers, data, repositories☆99Nov 15, 2024Updated last year
- ☆252Dec 2, 2024Updated last year
- The Prodigy optimizer and its variants for training neural networks.☆451Jan 16, 2025Updated last year
- ☆24May 6, 2025Updated 10 months ago
- Fast, Modern, and Low Precision PyTorch Optimizers☆125Dec 29, 2025Updated 2 months ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆218Apr 4, 2021Updated 4 years ago
- Schedule-Free Optimization in PyTorch☆2,257May 21, 2025Updated 9 months ago
- Test pytorch code with minimal computational overhead☆26Jun 8, 2023Updated 2 years ago
- Landing Page for Divide and Remaster v3☆25Jul 29, 2025Updated 7 months ago
- A C++/Cython audio limiter for Python.☆25Feb 8, 2023Updated 3 years ago
- ☆23Jan 5, 2025Updated last year
- The AdEMAMix Optimizer: Better, Faster, Older.☆186Sep 12, 2024Updated last year
- APOLLO: SGD-like Memory, AdamW-level Performance; MLSys'25 Oustanding Paper Honorable Mention☆271Nov 29, 2025Updated 3 months ago
- KAIST. CS492 Algorithms for NP-hard Problems. Spring 2024.☆23Jun 4, 2024Updated last year
- Efficient optimizers☆285Dec 20, 2025Updated 2 months ago
- [ICDCS 2023] Evaluation and Optimization of Gradient Compression for Distributed Deep Learning☆10Apr 28, 2023Updated 2 years ago
- Code for my collection of predictors/classifiers/etc☆14Jul 18, 2024Updated last year
- Home Made Diffusion Models☆193Dec 9, 2025Updated 2 months ago
- The official implementation of Convergent Graph Solvers (CGS)☆21Feb 1, 2022Updated 4 years ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,412Updated this week
- ☆10Jan 26, 2021Updated 5 years ago