Wenhao Hu 8abdaee243 Add cpu adagrad (#1358) | 3 years ago | |
---|---|---|
.. | ||
cpu_adam.cpp | 8abdaee243 Add cpu adagrad (#1358) | 3 years ago |
fused_adam_frontend.cpp | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 years ago |
multi_tensor_adam.cu | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 years ago |
multi_tensor_apply.cuh | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 years ago |