← all repositories

jettify/pytorch-optimizer

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

3.2k stars Python ML Frameworks
pytorch-optimizer
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.