jettify/pytorch-optimizer
A PyTorch-compatible library providing implementations of 20+ advanced optimization algorithms for deep learning model training.

Velocity · 7d
+1.3
★ / day
Trend
→steady
star history
This repository provides a curated collection of optimization algorithms compatible with PyTorch’s optimizer module. It implements research-grade optimizers including DiffGrad, LAMB, AdaBelief, Adamod, and Lookahead, which are used to train neural networks. Users can swap standard PyTorch optimizers with these alternatives by changing only the import statement and class name, making experimental comparison straightforward.