Ross Wightman 0b5ae49251 Remove adafactorbv numpy dep, hack fix for loading optimizer state w/ half prec momentum (need better one)

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

..

__init__.py

An impl of adafactor as per big vision (scaling vit) changes

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

adabelief.py

…

adafactor_bv.py

Remove adafactorbv numpy dep, hack fix for loading optimizer state w/ half prec momentum (need better one)

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

adafactor.py

…

adahessian.py

…

adamp.py

…

adamw.py

…

adan.py

…

lamb.py

…

lars.py

…

lion.py

…

lookahead.py

…

madgrad.py

…

nadam.py

…

nadamw.py

…

nvnovograd.py

…

optim_factory.py

An impl of adafactor as per big vision (scaling vit) changes

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

radam.py

…

rmsprop_tf.py

…

sgdp.py

…

sgdw.py

…