Search 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

submodlib-py

Summarize Massive Datasets using Submodular Optimization

labml-nn

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

opytimizer

Nature-Inspired Python Optimizer

cgds

Package for CGD and ACGD optimizers

torchzero

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

rose-opt

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

keras-adamw

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

keras-radam

RAdam implemented in Keras & TensorFlow

keras-rectified-adam

RAdam implemented in Keras & TensorFlow

rangerlite

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

keras-lookahead

Lookahead mechanism for optimizers in Keras

gradient-centralization-tf

Instantly improve your optimizer with just 2 lines of code!

tensorflow-radam

RAdam implemented in Keras & TensorFlow

lion-tf

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

optiml

Optimizers for/and sklearn compatible Machine Learning models