.. |
adagrad
|
a10e4811fe
force set lf instead of crlf (https://github.com/pre-commit/pre-commit-hooks#mixed-line-ending) (#1598)
|
2 years ago |
adam
|
a10e4811fe
force set lf instead of crlf (https://github.com/pre-commit/pre-commit-hooks#mixed-line-ending) (#1598)
|
2 years ago |
aio
|
316c4a43e0
Add flake8 to pre-commit checks (#2051)
|
2 years ago |
common
|
a10e4811fe
force set lf instead of crlf (https://github.com/pre-commit/pre-commit-hooks#mixed-line-ending) (#1598)
|
2 years ago |
includes
|
c3001324b4
Add predicated global load (#2373)
|
2 years ago |
lamb
|
2e3769a1f4
Enable fused_lamb_cuda_kernel on ROCm (#2148)
|
2 years ago |
quantization
|
954e0c61f1
mem access for quantize kernel (#2331)
|
2 years ago |
sparse_attention
|
74493b2bee
support CPU Adam and Adagrad on Windows with SDK 10.0.22000 (#1634)
|
2 years ago |
transformer
|
ec13da6ba7
add SD injection policy (#2381)
|
2 years ago |
utils
|
31f46feee2
DeepSpeed JIT op + PyPI support (#496)
|
4 years ago |