Ross Wightman 490d222dd8 Fix issue taking device from V before V exists

2025-01-31 12:52:47 -08:00

..

__init__.py

…

_optim_factory.py

Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now.

2025-01-27 21:02:26 -08:00

_param_groups.py

…

_types.py

…

adabelief.py

Fix typos

2025-01-19 13:39:40 -08:00

adafactor_bv.py

…

adafactor.py

…

adahessian.py

Fix typos

2025-01-19 13:39:40 -08:00

adamp.py

…

adamw.py

…

adan.py

Add caution to Adan. Add decouple decay option to LAMB.

2024-12-05 13:50:30 -08:00

adopt.py

…

kron.py

Fix issue taking device from V before V exists

2025-01-31 12:52:47 -08:00

lamb.py

Add caution to Adan. Add decouple decay option to LAMB.

2024-12-05 13:50:30 -08:00

laprop.py

…

lars.py

…

lion.py

…

lookahead.py

…

madgrad.py

…

mars.py

Add caution to Adan. Add decouple decay option to LAMB.

2024-12-05 13:50:30 -08:00

nadam.py

…

nadamw.py

…

nvnovograd.py

…

optim_factory.py

…

radam.py

…

rmsprop_tf.py

…

sgdp.py

…

sgdw.py

…