bitsandbytes-rocm/bitsandbytes/optim
2023-03-09 11:54:54 -08:00
..
__init__.py initial commit, slowly work from interface into the kernel 2023-03-09 08:08:46 -08:00
adagrad.py
adam.py
adamw.py
lamb.py
lars.py
lion.py use epsilon as beta2 for lion, complete most of the logic in kernel.cu for all functions 2023-03-09 11:54:54 -08:00
optimizer.py
rmsprop.py
sgd.py