Ross Wightman e6d72ed1b7 Update adafactor comments / attrib

2024-11-12 09:30:26 -08:00

..

__init__.py

Add ADOPT optimizer

2024-11-08 09:10:17 -08:00

adabelief.py

…

adafactor_bv.py

Update adafactor comments / attrib

2024-11-12 09:30:26 -08:00

adafactor.py

Update adafactor comments / attrib

2024-11-12 09:30:26 -08:00

adahessian.py

…

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

…

adan.py

…

adopt.py

Remove an indent level in init_group for adopt, update optim tests, adopt failing rosenbrock

2024-11-08 10:38:42 -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-07 21:42:24 -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-07 21:42:24 -08:00

lion.py

…

lookahead.py

…

madgrad.py

…

nadam.py

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

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

nadamw.py

…

nvnovograd.py

…

optim_factory.py

Add ADOPT optimizer

2024-11-08 09:10:17 -08:00

radam.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

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

…

sgdw.py

…