.. |
__init__.py
|
c3c8d5dd93
AMD support (#1430)
|
2 年之前 |
async_io.py
|
91e15593ea
Control ds_report output (#1622)
|
2 年之前 |
builder.py
|
73b5d9833a
Fix setup.py crash when torch is not installed. (#1866)
|
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
|
c3c8d5dd93
AMD support (#1430)
|
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
|
8164ea9e6d
Fixing several bugs in the inference-api and the kernels (#1951)
|
2 年之前 |
utils.py
|
599258f979
ZeRO 3 Offload (#834)
|
3 年之前 |