bitsandbytes-rocm/bitsandbytes/optim
2022-07-01 17:25:44 +03:00
..
__init__.py Fix imports 2022-07-01 17:25:44 +03:00
adagrad.py Remove unused imports, fix NotImplementedError 2022-06-30 18:14:20 +03:00
adam.py Merge branch 'main' of github.com:facebookresearch/bitsandbytes into 0.26.0 2021-11-29 08:24:17 -08:00
adamw.py Remove unused imports, fix NotImplementedError 2022-06-30 18:14:20 +03:00
lamb.py
lars.py Remove unused imports, fix NotImplementedError 2022-06-30 18:14:20 +03:00
optimizer.py Added module override, bnb.nn.Embedding #13 #15 #19 2021-11-29 09:32:13 -08:00
rmsprop.py Add a CPU-only build option 2022-07-01 17:16:10 +03:00
sgd.py Remove unused imports, fix NotImplementedError 2022-06-30 18:14:20 +03:00