Optimizers Python Packages

lion-pytorch

🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch

prodigy-plus-schedule-free

Prodigy and Schedule-Free, together at last.

torch-optimi

Fast, Modern, and Low Precision PyTorch Optimizers

adafactor8bit

8-bit Adafactor Optimizer with Fused CUDA Kernels

rose-opt

🌹 Rose: Range-Of-Slice Equilibration PyTorch optimizer. Stateless optimization through range-normalized gradient updates.

opytimizer

🐦 Opytimizer is a Python library consisting of meta-heuristic optimization algorithms.

submodlib-py

Summarize Massive Datasets using Submodular Optimization

labml-nn

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

torchzero

Modular optimization library for PyTorch (work-in-progress).

cgds

Adaptive Competitive Gradient Descent optimizer

zij

A canon of deep learning optimizers.

optiml

Optimizers for/and sklearn compatible Machine Learning models

keras-radam

RAdam implemented in Keras & TensorFlow

keras-rectified-adam

RAdam implemented in Keras & TensorFlow

keras-adamw

Keras implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers

keras-lookahead

Lookahead mechanism for optimizers in Keras

tensorflow-radam

RAdam implemented in Keras & TensorFlow

rangerlite

A pytorch implementation of the Lookahead optimizer with RAdam as inner optimizer

gradient-centralization-tf

Implement Gradient Centralization in TensorFlow

lion-tf

Lion - EvoLved Sign Momentum w/ New Optimizer API in TensorFlow 2.11+