.. |
__init__.py
|
c3c8d5dd93
AMD support (#1430)
|
2 年之前 |
async_io.py
|
91e15593ea
Control ds_report output (#1622)
|
2 年之前 |
builder.py
|
a84f9da8a2
add cuda 11.7 (#2211)
|
2 年之前 |
cpu_adagrad.py
|
c3c8d5dd93
AMD support (#1430)
|
2 年之前 |
cpu_adam.py
|
737fee63ba
Update cpu_adam.py (#1968)
|
2 年之前 |
fused_adam.py
|
73b5d9833a
Fix setup.py crash when torch is not installed. (#1866)
|
2 年之前 |
fused_lamb.py
|
73b5d9833a
Fix setup.py crash when torch is not installed. (#1866)
|
2 年之前 |
quantizer.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
sparse_attn.py
|
316c4a43e0
Add flake8 to pre-commit checks (#2051)
|
2 年之前 |
stochastic_transformer.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
transformer.py
|
b4fcd98ff0
Inference PP changes for neox (#1899)
|
2 年之前 |
transformer_inference.py
|
8b2a63717a
Add support of OPT models (#2205)
|
2 年之前 |
utils.py
|
599258f979
ZeRO 3 Offload (#834)
|
3 年之前 |