Ross Wightman 79abc25f55 Add ADOPT optimizer

2024-11-12 20:49:01 -08:00

..

__init__.py

Add ADOPT optimizer

2024-11-12 20:49:01 -08:00

adabelief.py

More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.

2021-08-18 17:21:56 -07:00

adafactor_bv.py

Improve row/col dim var name

2024-11-12 20:49:01 -08:00

adafactor.py

Cleanup original adafactor impl, add row/col dim heuristic that works with both conv and linear layers

2024-11-12 20:49:01 -08:00

adahessian.py

Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.

2020-10-09 17:24:43 -07:00

adamp.py

More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.

2021-08-18 17:21:56 -07:00

adamw.py

More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.

2021-08-18 17:21:56 -07:00

adan.py

Add Adan optimizer

2022-09-23 16:05:52 -07:00

adopt.py

Add ADOPT optimizer

2024-11-12 20:49:01 -08:00

lamb.py

A bit of lars/lamb cleanup, torch.where supports scalars properly now, make lamb grad clipping optional, clean it up a bit

2024-11-12 20:49:01 -08:00

lars.py

A bit of lars/lamb cleanup, torch.where supports scalars properly now, make lamb grad clipping optional, clean it up a bit

2024-11-12 20:49:01 -08:00

lion.py

…

lookahead.py

Fix Pytorch 2.0 breakage for Lookahead optimizer adapter

2023-06-02 08:39:07 -07:00

madgrad.py

MOAR optimizer changes. Woo!

2021-08-18 22:20:35 -07:00

nadam.py

Add NadamW based on mlcommons algorithm, added multi-tensor step

2023-06-13 20:45:17 -07:00

nadamw.py

…

nvnovograd.py

More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.

2021-08-18 17:21:56 -07:00

optim_factory.py

Add ADOPT optimizer

2024-11-12 20:49:01 -08:00

radam.py

…

rmsprop_tf.py

More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.

2021-08-18 17:21:56 -07:00

sgdp.py

More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.

2021-08-18 17:21:56 -07:00

sgdw.py

Update sgdw for older pytorch

2023-12-11 12:10:29 -08:00